home *** CD-ROM | disk | FTP | other *** search
/ Maximum CD 2011 June / maximum-cd-2011-06.iso / DiscContents / LibO_3.3.1_Win_x86_install_multi.exe / libreoffice1.cab / time_hashlib.py < prev    next >
Encoding:
Python Source  |  2011-02-15  |  2.8 KB  |  88 lines
  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.