Made test work in python3

2018-01-15 22:29:37 -05:00 · 2018-01-15 22:29:37 -05:00 · 1e051f1a60
parent 077a3b60e7
commit 1e051f1a60
1 changed files with 6 additions and 1 deletions
--- 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