@@ -1884,10 +1884,11 @@ def test_contains(self):
1884
1884
self .assertFalse (0 in ci )
1885
1885
self .assertFalse (1 in ci )
1886
1886
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 ])
1888
1889
self .assertFalse (np .nan in ci )
1889
1890
1890
- ci = CategoricalIndex (list ('aabbca' ) + [np .nan ], categories = list ('cabdef' ) + [ np . nan ] )
1891
+ ci = CategoricalIndex (list ('aabbca' ) + [np .nan ], categories = list ('cabdef' ))
1891
1892
self .assertTrue (np .nan in ci )
1892
1893
1893
1894
def test_min_max (self ):
@@ -2052,7 +2053,7 @@ def test_repr_roundtrip(self):
2052
2053
2053
2054
def test_isin (self ):
2054
2055
2055
- ci = CategoricalIndex (list ('aabca' ) + [np .nan ],categories = ['c' ,'a' ,'b' , np . nan ])
2056
+ ci = CategoricalIndex (list ('aabca' ) + [np .nan ],categories = ['c' ,'a' ,'b' ])
2056
2057
tm .assert_numpy_array_equal (ci .isin (['c' ]),np .array ([False ,False ,False ,True ,False ,False ]))
2057
2058
tm .assert_numpy_array_equal (ci .isin (['c' ,'a' ,'b' ]),np .array ([True ]* 5 + [False ]))
2058
2059
tm .assert_numpy_array_equal (ci .isin (['c' ,'a' ,'b' ,np .nan ]),np .array ([True ]* 6 ))
@@ -2100,10 +2101,11 @@ def test_equals(self):
2100
2101
# tests
2101
2102
# make sure that we are testing for category inclusion properly
2102
2103
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' )))
2104
2106
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 ]))
2107
2109
2108
2110
2109
2111
class Numeric (Base ):
0 commit comments