Rework Python encode/decode benchmarks to work with harness

haskell · Jan 9, 2014 · 8cc212a · 8cc212a
1 parent 9636fda
commit 8cc212a
Show file tree

Hide file tree

Showing 2 changed files with 22 additions and 2 deletions.
diff --git a/benchmarks/encode.py b/benchmarks/encode.py
@@ -2,6 +2,16 @@
 
 import json, sys, time
 
+def isint(x):
+    try:
+        int(x)
+        return True
+    except:
+        return False
+
+if len(sys.argv) > 2 and isint(sys.argv[1]) and isint(sys.argv[2]):
+    sys.argv.pop(1)
+
 count = int(sys.argv[1])
 
 for n in sys.argv[2:]:
@@ -11,4 +21,4 @@
     for i in xrange(count):
         json.dumps(obj)
     end = time.time()
-    print ' ', end - start
+    print '  %d good, %gs' % (count, end - start)
diff --git a/benchmarks/parse.py b/benchmarks/parse.py
@@ -2,6 +2,16 @@
 
 import json, sys, time
 
+def isint(x):
+    try:
+        int(x)
+        return True
+    except:
+        return False
+
+if len(sys.argv) > 2 and isint(sys.argv[1]) and isint(sys.argv[2]):
+    sys.argv.pop(1)
+
 count = int(sys.argv[1])
 
 for n in sys.argv[2:]:
@@ -12,4 +22,4 @@
         fp.seek(0)
         val = json.load(fp)
     end = time.time()
-    print ' ', end - start
+    print '  %d good, %gs' % (count, end - start)