@@ -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
21092111class Numeric (Base ):
0 commit comments