| 12345678910111213141516171819202122232425 |
- # -*- coding: utf-8 -*-
- from pandas import Categorical
- import pandas.util.testing as tm
- class TestCategoricalSubclassing(object):
- def test_constructor(self):
- sc = tm.SubclassedCategorical(['a', 'b', 'c'])
- assert isinstance(sc, tm.SubclassedCategorical)
- tm.assert_categorical_equal(sc, Categorical(['a', 'b', 'c']))
- def test_from_codes(self):
- sc = tm.SubclassedCategorical.from_codes([1, 0, 2], ['a', 'b', 'c'])
- assert isinstance(sc, tm.SubclassedCategorical)
- exp = Categorical.from_codes([1, 0, 2], ['a', 'b', 'c'])
- tm.assert_categorical_equal(sc, exp)
- def test_map(self):
- sc = tm.SubclassedCategorical(['a', 'b', 'c'])
- res = sc.map(lambda x: x.upper())
- assert isinstance(res, tm.SubclassedCategorical)
- exp = Categorical(['A', 'B', 'C'])
- tm.assert_categorical_equal(res, exp)
|
