pyca · Ivoz · Oct 6, 2013 · Oct 8, 2013
diff --git a/profile.py b/profile.py
@@ -0,0 +1,55 @@
+import os
+import sys
+import cProfile
+
+import ed25519
+
+help_desc = """
+profile.py [case] [loops]
+
+Run profiling of ed25519 functions
+
+case  - must be one of:  [default: val]
+        pub -- generating a public key
+        sig -- generating a signature
+        val -- validating a signature
+
+loops - how many times to repeat test  [default: 300]
+"""
+
+seed = os.urandom(32)
+data = b"The quick brown fox jumps over the lazy dog" * 20
+private_key = seed
+public_key = ed25519.publickey(seed)
+signature = ed25519.signature(data, private_key, public_key)
+
+gen_public_key = 'ed25519.publickey(seed)'
+gen_signature = 'ed25519.signature(data, private_key, public_key)'
+do_validation = 'ed25519.checkvalid(signature, data, public_key)'
+
+case = {
+    'pub': gen_public_key,
+    'sig': gen_signature,
+    'val': do_validation,
+}
+
+length = 300
+choice = ''
+
+if __name__ == "__main__":
+
+    if len(sys.argv) >= 2:
+        if sys.argv[1] == '-h' or sys.argv[1] == '--help':
+            print(help_desc)
+            exit()
+        choice = sys.argv[1]
+    if len(sys.argv) >= 3:
+        length = int(sys.argv[2])
+
+    method = case.get(choice, 'val')
+
+    loop = '[%s for _ in range(%d)]'
+
+    print('Running %s, %d times\n' % (case[method], length))
+
+    cProfile.run(loop % (case[method], length), sort='time')