Home | History | Annotate | Download | only in test 
      1 # It's intended that this script be run by hand.  It runs speed tests on
      2 # hashlib functions; it does not test for correctness.
      3 
      4 import sys, time
      5 import hashlib
      6 
      7 
      8 def creatorFunc():
      9     raise RuntimeError, "eek, creatorFunc not overridden"
     10 
     11 def test_scaled_msg(scale, name):
     12     iterations = 106201/scale * 20
     13     longStr = 'Z'*scale
     14 
     15     localCF = creatorFunc
     16     start = time.time()
     17     for f in xrange(iterations):
     18         x = localCF(longStr).digest()
     19     end = time.time()
     20 
     21     print ('%2.2f' % (end-start)), "seconds", iterations, "x", len(longStr), "bytes", name
     22 
     23 def test_create():
     24     start = time.time()
     25     for f in xrange(20000):
     26         d = creatorFunc()
     27     end = time.time()
     28 
     29     print ('%2.2f' % (end-start)), "seconds", '[20000 creations]'
     30 
     31 def test_zero():
     32     start = time.time()
     33     for f in xrange(20000):
     34         x = creatorFunc().digest()
     35     end = time.time()
     36 
     37     print ('%2.2f' % (end-start)), "seconds", '[20000 "" digests]'
     38 
     39 
     40 
     41 hName = sys.argv[1]
     42 
     43 #
     44 # setup our creatorFunc to test the requested hash
     45 #
     46 if hName in ('_md5', '_sha'):
     47     exec 'import '+hName
     48     exec 'creatorFunc = '+hName+'.new'
     49     print "testing speed of old", hName, "legacy interface"
     50 elif hName == '_hashlib' and len(sys.argv) > 3:
     51     import _hashlib
     52     exec 'creatorFunc = _hashlib.%s' % sys.argv[2]
     53     print "testing speed of _hashlib.%s" % sys.argv[2], getattr(_hashlib, sys.argv[2])
     54 elif hName == '_hashlib' and len(sys.argv) == 3:
     55     import _hashlib
     56     exec 'creatorFunc = lambda x=_hashlib.new : x(%r)' % sys.argv[2]
     57     print "testing speed of _hashlib.new(%r)" % sys.argv[2]
     58 elif hasattr(hashlib, hName) and callable(getattr(hashlib, hName)):
     59     creatorFunc = getattr(hashlib, hName)
     60     print "testing speed of hashlib."+hName, getattr(hashlib, hName)
     61 else:
     62     exec "creatorFunc = lambda x=hashlib.new : x(%r)" % hName
     63     print "testing speed of hashlib.new(%r)" % hName
     64 
     65 try:
     66     test_create()
     67 except ValueError:
     68     print
     69     print "pass argument(s) naming the hash to run a speed test on:"
     70     print " '_md5' and '_sha' test the legacy builtin md5 and sha"
     71     print " '_hashlib' 'openssl_hName' 'fast' tests the builtin _hashlib"
     72     print " '_hashlib' 'hName' tests builtin _hashlib.new(shaFOO)"
     73     print " 'hName' tests the hashlib.hName() implementation if it exists"
     74     print "         otherwise it uses hashlib.new(hName)."
     75     print
     76     raise
     77 
     78 test_zero()
     79 test_scaled_msg(scale=106201, name='[huge data]')
     80 test_scaled_msg(scale=10620, name='[large data]')
     81 test_scaled_msg(scale=1062, name='[medium data]')
     82 test_scaled_msg(scale=424, name='[4*small data]')
     83 test_scaled_msg(scale=336, name='[3*small data]')
     84 test_scaled_msg(scale=212, name='[2*small data]')
     85 test_scaled_msg(scale=106, name='[small data]')
     86 test_scaled_msg(scale=creatorFunc().digest_size, name='[digest_size data]')
     87 test_scaled_msg(scale=10, name='[tiny data]')
     88