From b609e07bbdb6b0e7a06d5605ae30f342c40f30fe Mon Sep 17 00:00:00 2001
From: Jeff Reback <jeff@reback.net>
Date: Mon, 21 May 2018 20:33:49 -0400
Subject: [PATCH] PY2 div compat

---
 pandas/tests/extension/integer/test_integer.py | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/pandas/tests/extension/integer/test_integer.py b/pandas/tests/extension/integer/test_integer.py
index 39526f6e18b5a5..055368e78efe01 100644
--- 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