|
2 | 2 | import pandas._testing as tm
|
3 | 3 |
|
4 | 4 |
|
| 5 | +class SubclassedCategorical(Categorical): |
| 6 | + pass |
| 7 | + |
| 8 | + |
5 | 9 | class TestCategoricalSubclassing:
|
6 | 10 | def test_constructor(self):
|
7 |
| - sc = tm.SubclassedCategorical(["a", "b", "c"]) |
8 |
| - assert isinstance(sc, tm.SubclassedCategorical) |
| 11 | + sc = SubclassedCategorical(["a", "b", "c"]) |
| 12 | + assert isinstance(sc, SubclassedCategorical) |
9 | 13 | tm.assert_categorical_equal(sc, Categorical(["a", "b", "c"]))
|
10 | 14 |
|
11 | 15 | def test_from_codes(self):
|
12 |
| - sc = tm.SubclassedCategorical.from_codes([1, 0, 2], ["a", "b", "c"]) |
13 |
| - assert isinstance(sc, tm.SubclassedCategorical) |
| 16 | + sc = SubclassedCategorical.from_codes([1, 0, 2], ["a", "b", "c"]) |
| 17 | + assert isinstance(sc, SubclassedCategorical) |
14 | 18 | exp = Categorical.from_codes([1, 0, 2], ["a", "b", "c"])
|
15 | 19 | tm.assert_categorical_equal(sc, exp)
|
16 | 20 |
|
17 | 21 | def test_map(self):
|
18 |
| - sc = tm.SubclassedCategorical(["a", "b", "c"]) |
| 22 | + sc = SubclassedCategorical(["a", "b", "c"]) |
19 | 23 | res = sc.map(lambda x: x.upper(), na_action=None)
|
20 |
| - assert isinstance(res, tm.SubclassedCategorical) |
| 24 | + assert isinstance(res, SubclassedCategorical) |
21 | 25 | exp = Categorical(["A", "B", "C"])
|
22 | 26 | tm.assert_categorical_equal(res, exp)
|
0 commit comments