Skip to content

Commit 5ef161f

Browse files
committed
Add test fix for Travis CI python different versions
1 parent 43bb299 commit 5ef161f

File tree

2 files changed

+18
-5
lines changed

2 files changed

+18
-5
lines changed

requirements_testing.txt

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -2,3 +2,4 @@
22
nose
33
coverage
44
codacy-coverage
5+
six

tests/test_performance.py

Lines changed: 17 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,7 @@
11
import math
2-
import timeit
2+
from timeit import default_timer as timer
3+
4+
import six
35

46
import dist
57

@@ -23,10 +25,20 @@ def test_positive_scenario():
2325

2426

2527
def test_performance():
26-
pure_py_time = timeit.timeit("pure_py_dist(10.1, 12.1, 10.1, 10.1)", number=10000000,
27-
setup="from __main__ import pure_py_dist")
28-
ext_time = timeit.timeit("dist.compute(10.1, 12.1, 10.1, 10.1)", number=10000000,
29-
setup="from __main__ import dist")
28+
start_time = timer()
29+
30+
for _ in six.xrange(10000000):
31+
pure_py_dist(10.1, 12.1, 10.1, 10.1)
32+
33+
pure_py_time = timer() - start_time
34+
35+
start_time = timer()
36+
37+
for _ in six.xrange(10000000):
38+
dist.compute(10.1, 12.1, 10.1, 10.1)
39+
40+
ext_time = timer() - start_time
41+
3042
print("ext_time: {} pure_py_time {}".format(pure_py_time, ext_time))
3143
# My Python/C API Extension is in ~ 3 times faster than fast pure python implementation
3244
assert ext_time < pure_py_time

0 commit comments

Comments
 (0)