numpy.int64 is not int on py3

biosustain · Sep 4, 2015 · 1f9dec0 · 1f9dec0
1 parent 8762cb5
commit 1f9dec0
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/tests/test_strain_design_heuristics.py b/tests/test_strain_design_heuristics.py
@@ -21,6 +21,7 @@
 import inspyred
 import pickle
 from inspyred.ec import Bounder
+import numpy
 from ordered_set import OrderedSet
 
 from pandas.util.testing import assert_frame_equal
@@ -441,7 +442,7 @@ def test_fixed_size_linear_set_generator(self):
         for _ in range(10000):
             candidate = linear_set_generator(self.random, self.args)
             for c in candidate:
-                self.assertIsInstance(c[0], six.integer_types)
+                self.assertIsInstance(c[0], (int, numpy.int64, numpy.int32))
                 self.assertIsInstance(c[1], float)
 
             self.assertLessEqual(len(candidate), 10)