Make unit tests work with latest version of pypy

aholmberg · aholmberg · commit 2dba082da6f7 · 2015-03-06T08:48:34.000-06:00
diff --git a/tests/unit/test_policies.py b/tests/unit/test_policies.py
@@ -140,20 +140,21 @@ def host_down():
 
         # make the GIL switch after every instruction, maximizing
         # the chace of race conditions
-        if six.PY2:
+        check = six.PY2 or '__pypy__' in sys.builtin_module_names
+        if check:
             original_interval = sys.getcheckinterval()
         else:
             original_interval = sys.getswitchinterval()
 
         try:
-            if six.PY2:
+            if check:
                 sys.setcheckinterval(0)
             else:
                 sys.setswitchinterval(0.0001)
             map(lambda t: t.start(), threads)
             map(lambda t: t.join(), threads)
         finally:
-            if six.PY2:
+            if check:
                 sys.setcheckinterval(original_interval)
             else:
                 sys.setswitchinterval(original_interval)
diff --git a/tests/unit/test_types.py b/tests/unit/test_types.py
@@ -20,6 +20,7 @@
 import calendar
 import datetime
 import tempfile
+import six
 import time
 
 import cassandra
@@ -228,7 +229,7 @@ def __init__(self, subtypes, names):
 
             @classmethod
             def apply_parameters(cls, subtypes, names):
-                return cls(subtypes, [unhexlify(name) if name is not None else name for name in names])
+                return cls(subtypes, [unhexlify(six.b(name)) if name is not None else name for name in names])
 
         class BarType(FooType):
             typename = 'org.apache.cassandra.db.marshal.BarType'
