1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
| import numpy as np
|
| from pandas import (
| Categorical,
| Series,
| )
| import pandas._testing as tm
|
|
| class TestCategoricalComparisons:
| def test_categorical_nan_equality(self):
| cat = Series(Categorical(["a", "b", "c", np.nan]))
| expected = Series([True, True, True, False])
| result = cat == cat
| tm.assert_series_equal(result, expected)
|
| def test_categorical_tuple_equality(self):
| # GH 18050
| ser = Series([(0, 0), (0, 1), (0, 0), (1, 0), (1, 1)])
| expected = Series([True, False, True, False, False])
| result = ser == (0, 0)
| tm.assert_series_equal(result, expected)
|
| result = ser.astype("category") == (0, 0)
| tm.assert_series_equal(result, expected)
|
|