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.
9 raise RuntimeError, "eek, creatorFunc not overridden"
11 def test_scaled_msg(scale
, name
):
12 iterations
= 106201/scale
* 20
17 for f
in xrange(iterations
):
18 x
= localCF(longStr
).digest()
21 print ('%2.2f' % (end
-start
)), "seconds", iterations
, "x", len(longStr
), "bytes", name
25 for f
in xrange(20000):
29 print ('%2.2f' % (end
-start
)), "seconds", '[20000 creations]'
33 for f
in xrange(20000):
34 x
= creatorFunc().digest()
37 print ('%2.2f' % (end
-start
)), "seconds", '[20000 "" digests]'
44 # setup our creatorFunc to test the requested hash
46 if hName
in ('_md5', '_sha'):
48 exec 'creatorFunc = '+hName
+'.new'
49 print "testing speed of old", hName
, "legacy interface"
50 elif hName
== '_hashlib' and len(sys
.argv
) > 3:
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:
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
)
62 exec "creatorFunc = lambda x=hashlib.new : x(%r)" % hName
63 print "testing speed of hashlib.new(%r)" % hName
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)."
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]')