Skip to content

Commit b609e07

Browse files
committed
PY2 div compat
1 parent 0758f1d commit b609e07

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

pandas/tests/extension/integer/test_integer.py

+3-2
Original file line numberDiff line numberDiff line change
@@ -4,6 +4,7 @@
44
import pytest
55

66
from pandas.tests.extension import base
7+
from pandas.compat import PY2
78

89
from pandas.core.arrays import (
910
to_integer_array,
@@ -143,8 +144,8 @@ def compare(self, s, op, other):
143144
rs = pd.Series(s.values.data)
144145
expected = getattr(rs, op)(other)
145146

146-
# truediv can make infs
147-
if 'truediv' in op:
147+
# truediv/div can make infs
148+
if 'truediv' in op or (PY2 and 'div' in op):
148149
fill_value = np.nan
149150
else:
150151
fill_value = 0

0 commit comments

Comments
 (0)