blob: b1b2871b3d05906c214128d73065a85ea2549786 [file] [log] [blame]
Sybren A. Stüvel360d0422011-08-10 12:52:59 +02001#!/usr/bin/env python
2
3import time
4import rsa
5
6poolsize = 8
7accurate = True
8
9def run_speed_test(bitsize):
10
11 iterations = 0
12 start = end = time.time()
13
14 # At least a number of iterations, and at least 2 seconds
15 while iterations < 10 or end - start < 2:
16 iterations += 1
17 rsa.newkeys(bitsize, accurate=accurate, poolsize=poolsize)
18 end = time.time()
19
20 duration = end - start
21 dur_per_call = duration / iterations
22
23 print '%5i bit: %9.3f sec. (%i iterations over %.1f seconds)' % (bitsize,
24 dur_per_call, iterations, duration)
25
26for bitsize in (128, 256, 384, 512, 1024, 2048, 3072, 4096):
27 run_speed_test(bitsize)
28
29