Merge pull request #237 from minrk/bench

benchmark fixes
pytoolz · Apr 20, 2015 · e2f10e6 · e2f10e6
2 parents 638f48c + 5f79d7d
commit e2f10e6
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 6 deletions.
diff --git a/bench/test_frequencies.py b/bench/test_frequencies.py
@@ -1,8 +1,8 @@
 from toolz import frequencies, identity
 
 
-big_data = range(1000)*1000
-small_data = range(100)
+big_data = list(range(1000)) * 1000
+small_data = list(range(100))
 
 
 def test_frequencies():

diff --git a/bench/test_groupby.py b/bench/test_groupby.py
@@ -1,7 +1,7 @@
 from toolz import groupby, identity
 
 
-data = range(1000)*1000
+data = list(range(1000)) * 1000
 
 
 def test_groupby():

diff --git a/bench/test_join.py b/bench/test_join.py
@@ -1,6 +1,10 @@
 from toolz.curried import *
 import random
 
+try:
+    xrange
+except NameError:
+    xrange = range
 
 def burn(seq):
     for item in seq:
@@ -12,20 +16,20 @@ def burn(seq):
 
 
 def test_many_to_many_large():
-    burn(join(get(0), identity, small, big))
+    burn(join(get(0), small, identity, big))
 
 
 def test_one_to_one_tiny():
     A = list(range(20))
     B = A[::2] + A[1::2][::-1]
 
     for i in xrange(50000):
-        burn(join(identity, identity, A, B))
+        burn(join(identity, A, identity, B))
 
 
 def test_one_to_many():
     A = list(range(20))
     B = pipe([20]*1000, map(range), concat, list)
 
     for i in xrange(100):
-        burn(join(identity, identity, A, B))
+        burn(join(identity, A, identity, B))