add 'setup.py speed'

warner · Jan 23, 2015 · 9767304 · 9767304
1 parent 74ad03e
commit 9767304
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 1 deletion.
diff --git a/.gitignore b/.gitignore
@@ -1,3 +1,3 @@
-/pake2/_version.py
 /MANIFEST
 /dist/
+/build/
diff --git a/setup.py b/setup.py
@@ -37,6 +37,21 @@ def do_test(self, which):
         return rc
 cmdclass["test"] = Test
 
+class Speed(Command):
+    description = "run speed benchmarks"
+    user_options = []
+    boolean_options = []
+    def initialize_options(self):
+        pass
+    def finalize_options(self):
+        pass
+    def run(self):
+        p = subprocess.Popen([sys.executable, "spake2/bench_spake2.py"])
+        rc = p.wait()
+        if rc != 0:
+            sys.exit(rc)
+cmdclass["speed"] = Speed
+
 setup(name="spake2",
       version=versioneer.get_version(),
       description="SPAKE2 password-authenticated key exchange (pure python)",