PY2 div compat

pandas-dev · May 22, 2018 · b609e07 · b609e07
1 parent 0758f1d
commit b609e07
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/pandas/tests/extension/integer/test_integer.py b/pandas/tests/extension/integer/test_integer.py
@@ -4,6 +4,7 @@
 import pytest
 
 from pandas.tests.extension import base
+from pandas.compat import PY2
 
 from pandas.core.arrays import (
     to_integer_array,
@@ -143,8 +144,8 @@ def compare(self, s, op, other):
         rs = pd.Series(s.values.data)
         expected = getattr(rs, op)(other)
 
-        # truediv can make infs
-        if 'truediv' in op:
+        # truediv/div can make infs
+        if 'truediv' in op or (PY2 and 'div' in op):
             fill_value = np.nan
         else:
             fill_value = 0