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]))
