[Python-checkins] r53059 - sandbox/trunk/import_in_py/importbench.py

[brett.cannon]

Author: brett.cannon
Date: Mon Dec 18 23:07:08 2006
New Revision: 53059

Modified:
   sandbox/trunk/import_in_py/importbench.py
Log:
Add tests for importing Python modules.

Also clean up output.


Modified: sandbox/trunk/import_in_py/importbench.py
==============================================================================
--- sandbox/trunk/import_in_py/importbench.py	(original)
+++ sandbox/trunk/import_in_py/importbench.py	Mon Dec 18 23:07:08 2006
@@ -41,11 +41,11 @@
         """Create requested files."""
         self.directory = tempfile.gettempdir()
         self.py_path = os.path.join(self.directory, self.module_name + '.py')
+        self.pyc_path = self.py_path + ('c' if __debug__ else 'o')
         with open(self.py_path, 'w') as py_file:
             py_file.write("# Temporary file for benchmarking import.")
         if self.pyc:
             compile_to_pyc(self.py_path, doraise=True)
-            self.pyc_path = self.py_path + ('c' if __debug__ else 'o')
         if not self.py:
             os.remove(self.py_path)
         sys.path.append(self.directory)
@@ -53,9 +53,9 @@
         
     def __exit__(self, *args):
         """Clean up created state."""
-        if self.py:
+        if os.path.exists(self.py_path):
             os.remove(self.py_path)
-        if self.pyc:
+        if os.path.exists(self.pyc_path):
             os.remove(self.pyc_path)
 
 
@@ -65,10 +65,10 @@
     sys.path = []
     sys.meta_path = []
     sys.path_importer_cache.clear()
-    with PyPycFiles() as py_pyc_state:
+    with PyPycFiles() as file_state:
         # Force the module into sys.modules .
-        __import__(py_pyc_state.module_name, {})
-        timer = Timer("__import__(%r, {}, {}, [], 0)" % py_pyc_state.module_name)
+        __import__(file_state.module_name, {})
+        timer = Timer("__import__(%r, {}, {}, [], 0)" % file_state.module_name)
         return timer.repeat(repeat, times)
 
 @save_import_state
@@ -90,14 +90,43 @@
         timer = Timer("try:__import__(%r, {}, {}, [], 0)\nexcept ImportError:pass" %
                         bad_name)
         return timer.repeat(repeat, times)
-    
-    
+
+def py_pyc_module_benchmark(py, pyc):
+    @save_import_state
+    def inner(times, repeat):
+        sys.path = []
+        sys.meta_path = []
+        with PyPycFiles(py, pyc) as file_state:
+            import_stmt = ("__import__(%(module)r, {}, {}, [], 0);"
+                            "del sys.modules[%(module)r]")
+            import_stmt = import_stmt % {'module':file_state.module_name}
+            # Warm up path_importer_cache.
+            exec import_stmt in globals(), {}
+            # Benchmark.
+            timer = Timer(import_stmt, "import sys")
+            return timer.repeat(repeat, times)
+    return inner
+     
+bench_pyc_with_py = py_pyc_module_benchmark(py=True, pyc=True)
+bench_pyc_without_py = py_pyc_module_benchmark(py=False, pyc=True)
+bench_py_without_pyc = py_pyc_module_benchmark(py=True, pyc=False)
+
+
 def main():
+    max_name_len = max(len(name) for name in globals().keys())
+    repetitions = 2
+    iterations = 10000
+    print ("Repeating tests %s times with %s iterations ..." %
+            (repetitions, iterations))
+    print
+    print "%s time per repetition (in ms)" % "benchmark".ljust(max_name_len)
+    print '-' * 50
     for name, item in globals().iteritems():
         if name.startswith('bench_'):
-            timings = item(10000, 1)
-            print "%s: %s" % (item.__name__, timings)
-            
+            timings = item(iterations, repetitions)
+            print "%s %s" % (name.ljust(max_name_len),
+                             [int(result*1000) for result in timings])
+
             
 if __name__ == '__main__':
     main()