def test_isin(self): ci = CategoricalIndex(list('aabca') + [np.nan], categories=['c', 'a', 'b']) tm.assert_numpy_array_equal( ci.isin(['c']), np.array([False, False, False, True, False, False])) tm.assert_numpy_array_equal(ci.isin(['c', 'a', 'b']), np.array([True] * 5 + [False])) tm.assert_numpy_array_equal(ci.isin(['c', 'a', 'b', np.nan]), np.array([True] * 6)) # mismatched categorical -> coerced to ndarray so doesn't matter tm.assert_numpy_array_equal( ci.isin(ci.set_categories(list('abcdefghi'))), np.array([True] * 6)) tm.assert_numpy_array_equal(ci.isin(ci.set_categories(list('defghi'))), np.array([False] * 5 + [True]))
def test_isin(self): ci = CategoricalIndex( list('aabca') + [np.nan], categories=['c', 'a', 'b']) tm.assert_numpy_array_equal( ci.isin(['c']), np.array([False, False, False, True, False, False])) tm.assert_numpy_array_equal( ci.isin(['c', 'a', 'b']), np.array([True] * 5 + [False])) tm.assert_numpy_array_equal( ci.isin(['c', 'a', 'b', np.nan]), np.array([True] * 6)) # mismatched categorical -> coerced to ndarray so doesn't matter tm.assert_numpy_array_equal( ci.isin(ci.set_categories(list('abcdefghi'))), np.array([True] * 6)) tm.assert_numpy_array_equal( ci.isin(ci.set_categories(list('defghi'))), np.array([False] * 5 + [True]))