Lib/test/time_hashlib.py

   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]')