Skip to content

Commit 7ca2190

Browse files
committed
COMPAT: remove deprecation warnings for nan in cat in test_index.py
1 parent 207efc2 commit 7ca2190

File tree

1 file changed

+8
-6
lines changed

1 file changed

+8
-6
lines changed

pandas/tests/test_index.py

+8-6
Original file line numberDiff line numberDiff line change
@@ -1884,10 +1884,11 @@ def test_contains(self):
18841884
self.assertFalse(0 in ci)
18851885
self.assertFalse(1 in ci)
18861886

1887-
ci = CategoricalIndex(list('aabbca'), categories=list('cabdef') + [np.nan])
1887+
with tm.assert_produces_warning(FutureWarning):
1888+
ci = CategoricalIndex(list('aabbca'), categories=list('cabdef') + [np.nan])
18881889
self.assertFalse(np.nan in ci)
18891890

1890-
ci = CategoricalIndex(list('aabbca') + [np.nan], categories=list('cabdef') + [np.nan])
1891+
ci = CategoricalIndex(list('aabbca') + [np.nan], categories=list('cabdef'))
18911892
self.assertTrue(np.nan in ci)
18921893

18931894
def test_min_max(self):
@@ -2052,7 +2053,7 @@ def test_repr_roundtrip(self):
20522053

20532054
def test_isin(self):
20542055

2055-
ci = CategoricalIndex(list('aabca') + [np.nan],categories=['c','a','b',np.nan])
2056+
ci = CategoricalIndex(list('aabca') + [np.nan],categories=['c','a','b'])
20562057
tm.assert_numpy_array_equal(ci.isin(['c']),np.array([False,False,False,True,False,False]))
20572058
tm.assert_numpy_array_equal(ci.isin(['c','a','b']),np.array([True]*5 + [False]))
20582059
tm.assert_numpy_array_equal(ci.isin(['c','a','b',np.nan]),np.array([True]*6))
@@ -2100,10 +2101,11 @@ def test_equals(self):
21002101
# tests
21012102
# make sure that we are testing for category inclusion properly
21022103
self.assertTrue(CategoricalIndex(list('aabca'),categories=['c','a','b']).equals(list('aabca')))
2103-
self.assertTrue(CategoricalIndex(list('aabca'),categories=['c','a','b',np.nan]).equals(list('aabca')))
2104+
with tm.assert_produces_warning(FutureWarning):
2105+
self.assertTrue(CategoricalIndex(list('aabca'),categories=['c','a','b',np.nan]).equals(list('aabca')))
21042106

2105-
self.assertFalse(CategoricalIndex(list('aabca') + [np.nan],categories=['c','a','b',np.nan]).equals(list('aabca')))
2106-
self.assertTrue(CategoricalIndex(list('aabca') + [np.nan],categories=['c','a','b',np.nan]).equals(list('aabca') + [np.nan]))
2107+
self.assertFalse(CategoricalIndex(list('aabca') + [np.nan],categories=['c','a','b']).equals(list('aabca')))
2108+
self.assertTrue(CategoricalIndex(list('aabca') + [np.nan],categories=['c','a','b']).equals(list('aabca') + [np.nan]))
21072109

21082110

21092111
class Numeric(Base):

0 commit comments

Comments
 (0)