BF: use safe Python 2 / 3 check for number

matthew-brett · matthew-brett · commit cd93c5b06d5b · 2018-01-15T15:44:01.000Z
Windows was barfing on long type at this check.

[skip ci]
diff --git a/nipy/algorithms/segmentation/tests/test_segmentation.py b/nipy/algorithms/segmentation/tests/test_segmentation.py
@@ -2,6 +2,8 @@
 """
 from __future__ import absolute_import
 
+from numbers import Number
+
 import numpy as np
 
 from nose.tools import assert_equal, assert_almost_equal
@@ -20,8 +22,8 @@
 
 
 def _check_dims(x, ndim, shape):
-    if isinstance(shape, int):
-        shape = (shape, )
+    if isinstance(shape, Number):
+        shape = (shape,)
     for i in range(ndim):
         assert_equal(x.shape[i], shape[i])
 
