import sys, time import hashlib from test import test_support def creatorFunc(): raise RuntimeError, "eek, creatorFunc not overridden" def test_scaled_msg(scale, name): iterations = 106201/scale * 20 longStr = 'Z'*scale localCF = creatorFunc start = time.time() for f in xrange(iterations): x = localCF(longStr).digest() end = time.time() print ('%2.2f' % (end-start)), "seconds", iterations, "x", len(longStr), "bytes", name def test_create(): start = time.time() for f in xrange(20000): d = creatorFunc() end = time.time() print ('%2.2f' % (end-start)), "seconds", '[20000 creations]' def test_zero(): start = time.time() for f in xrange(20000): x = creatorFunc().digest() end = time.time() print ('%2.2f' % (end-start)), "seconds", '[20000 "" digests]' ### this 'test' is not normally run. skip it if the test runner finds it if __name__ != '__main__': raise test_support.TestSkipped, "not a unit test (stand alone benchmark)" hName = sys.argv[1] # # setup our creatorFunc to test the requested hash # if hName in ('_md5', '_sha'): exec 'import '+hName exec 'creatorFunc = '+hName+'.new' print "testing speed of old", hName, "legacy interface" elif hName == '_hashlib' and len(sys.argv) > 3: import _hashlib exec 'creatorFunc = _hashlib.%s' % sys.argv[2] print "testing speed of _hashlib.%s" % sys.argv[2], getattr(_hashlib, sys.argv[2]) elif hName == '_hashlib' and len(sys.argv) == 3: import _hashlib exec 'creatorFunc = lambda x=_hashlib.new : x(%r)' % sys.argv[2] print "testing speed of _hashlib.new(%r)" % sys.argv[2] elif hasattr(hashlib, hName) and callable(getattr(hashlib, hName)): creatorFunc = getattr(hashlib, hName) print "testing speed of hashlib."+hName, getattr(hashlib, hName) else: exec "creatorFunc = lambda x=hashlib.new : x(%r)" % hName print "testing speed of hashlib.new(%r)" % hName try: test_create() except ValueError: print print "pass argument(s) naming the hash to run a speed test on:" print " '_md5' and '_sha' test the legacy builtin md5 and sha" print " '_hashlib' 'openssl_hName' 'fast' tests the builtin _hashlib" print " '_hashlib' 'hName' tests builtin _hashlib.new(shaFOO)" print " 'hName' tests the hashlib.hName() implementation if it exists" print " otherwise it uses hashlib.new(hName)." print raise test_zero() test_scaled_msg(scale=106201, name='[huge data]') test_scaled_msg(scale=10620, name='[large data]') test_scaled_msg(scale=1062, name='[medium data]') test_scaled_msg(scale=424, name='[4*small data]') test_scaled_msg(scale=336, name='[3*small data]') test_scaled_msg(scale=212, name='[2*small data]') test_scaled_msg(scale=106, name='[small data]') test_scaled_msg(scale=creatorFunc().digest_size, name='[digest_size data]') test_scaled_msg(scale=10, name='[tiny data]')