statprof profiling

vcf/test/prof.py

@@ -1,4 +1,4 @@
-import vcf as vcf
+import vcf
 import cProfile
 import timeit
 import pstats
@@ -20,5 +20,14 @@ def parse_1kg():
     n = 1
     t = timeit.timeit('parse_1kg()',  "from __main__ import parse_1kg", number=n)
     print t/n
+
+elif sys.argv[1] == 'stat':
+    import statprof
+    statprof.start()
+    try:
+        parse_1kg()
+    finally:
+        statprof.stop()
+        statprof.display()
 else:
     print 'prof.py profile/time'