MatrixPlayer
diff --git a/‎tests/parse.py‎
Lines changed: 19 additions & 8 deletions b/‎tests/parse.py‎
Lines changed: 19 additions & 8 deletions
diff --git a/‎tests/simple_parser.py‎
Lines changed: 0 additions & 30 deletions b/‎tests/simple_parser.py‎
Lines changed: 0 additions & 30 deletions
@@ -1,19 +1,30 @@
+import sys
+import os
 
 from test_parser import *
-import sys, os
 
 os.chdir(os.path.split(os.path.abspath(__file__))[0])
 sys.path.insert(0, os.path.abspath(os.path.join(os.pardir, "src")))
-
-
 import parser
 
 if __name__ == "__main__":
-    x = ""
+    p = parser.HTMLParser()
     if len(sys.argv) > 1:
         x = sys.argv[1]
+        if len(sys.argv) > 2:
+            import hotshot
+            import hotshot.stats
+            prof = hotshot.Profile('stats.prof')
+            prof.runcall(p.parse, x, False)
+            prof.close()
+            stats = hotshot.stats.load('stats.prof')
+            stats.strip_dirs()
+            stats.sort_stats('time')
+            stats.print_stats()
+        else:
+            document = p.parse(x)
+            print convertTreeDump(document.printTree())
     else:
-        x = "x"
-    p = parser.HTMLParser()
-    document = p.parse(StringIO.StringIO(x))
-    print convertTreeDump(document.printTree())
+        print """Pass one argument to parse the document and two to get an
+              indication on what's going on.
+              """