/
test.py
103 lines (73 loc) · 1.67 KB
/
test.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
# -------
# imports
# -------
import warnings
import glob
from itertools import count
from io import StringIO
from unittest import main, TestCase
from Decorator import fib, cache
import time
from random import randint
warnings.filterwarnings('ignore')
class TestCollatz (TestCase) :
# ----
# read
# ----
def test_basic (self) :
limit = 100
print("limit is " + str(limit))
x = range(limit)
arr = []
t1 = time.time()
for num in x:
# print("num is "+str(num))
arr += [fib(num)]
t2 = time.time()
print (t2 - t1)
fib2 = cache(fib)
arr2 = []
t1 = time.time()
for num in x:
# print("num is "+str(num))
arr2 += [fib2(num)]
t2 = time.time()
print (t2 - t1)
arr3 = []
t1 = time.time()
for num in x:
# print("num is "+str(num))
arr3 += [fib2(num)]
t2 = time.time()
print (t2 - t1)
# print(arr)
# print(arr2)
assert(arr == arr2)
assert(arr == arr3)
assert(arr3 == arr2)
def test_random (self) :
rand_arr = []
for i in range(50):
rand_arr += [randint(0, 10000)]
# rand_arr = [2, 5, 11, 8]
# print(rand_arr)
fib2 = cache(fib)
ans_arr1 = []
ans_arr2 = []
ans_arr3 = []
t1 = time.time()
for num in rand_arr:
ans_arr1 += [fib(num)]
t2 = time.time()
print (t2 - t1)
t1 = time.time()
for num in rand_arr:
ans_arr1 += [fib2(num)]
t2 = time.time()
print (t2 - t1)
t1 = time.time()
for num in rand_arr:
ans_arr1 += [fib2(num)]
t2 = time.time()
print (t2 - t1)
main()