RF: specify type of division for Py3 tests

BIDS · May 31, 2016 · 541740e · 541740e
1 parent 937c4cd
commit 541740e
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/datarray/tests/test_data_array.py b/datarray/tests/test_data_array.py
@@ -1,5 +1,8 @@
 '''Tests for DataArray and friend'''
 
+import sys
+PY3 = sys.version_info[0] >= 3
+
 import numpy as np
 
 from datarray.datarray import Axis, DataArray, NamedAxisError, \
@@ -446,7 +449,10 @@ def test_label_mismatch():
     nt.assert_raises(NamedAxisError, dar1.__add__, dar2)
     nt.assert_raises(NamedAxisError, dar1.__sub__, dar2)
     nt.assert_raises(NamedAxisError, dar1.__mul__, dar2)
-    nt.assert_raises(NamedAxisError, dar1.__div__, dar2)
+    nt.assert_raises(NamedAxisError, dar1.__floordiv__, dar2)
+    nt.assert_raises(NamedAxisError, dar1.__truediv__, dar2)
+    if not PY3:
+        nt.assert_raises(NamedAxisError, dar1.__div__, dar2)
 
 # -- Test DataArray.axes
 class TestAxesManager: