Made test work in python3

1e051f1a · Davis King · 077a3b60 · 1e051f1a
Commit 1e051f1a authored Jan 16, 2018 by Davis King
Hide whitespace changes
Inline Side-by-side

Showing with 6 additions and 1 deletion

test_array.py tools/python/test/test_array.py +6 -1

No files found.
--- a/tools/python/test/test_array.py
+++ b/tools/python/test/test_array.py
@@ -3,7 +3,12 @@ try:
    import cPickle as pickle  # Use cPickle on Python 2.7
 except ImportError:
    import pickle
-from types import FloatType
+
+try:
+    from types import FloatType
+except ImportError:
+    FloatType = float
+
 from pytest import raises