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
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
| """
| Tests for CategoricalIndex.__repr__ and related methods.
| """
| import pandas._config.config as cf
|
| from pandas import CategoricalIndex
|
|
| class TestCategoricalIndexRepr:
| def test_format_different_scalar_lengths(self):
| # GH#35439
| idx = CategoricalIndex(["aaaaaaaaa", "b"])
| expected = ["aaaaaaaaa", "b"]
| assert idx.format() == expected
|
| def test_string_categorical_index_repr(self):
| # short
| idx = CategoricalIndex(["a", "bb", "ccc"])
| expected = """CategoricalIndex(['a', 'bb', 'ccc'], categories=['a', 'bb', 'ccc'], ordered=False, dtype='category')""" # noqa:E501
| assert repr(idx) == expected
|
| # multiple lines
| idx = CategoricalIndex(["a", "bb", "ccc"] * 10)
| expected = """CategoricalIndex(['a', 'bb', 'ccc', 'a', 'bb', 'ccc', 'a', 'bb', 'ccc', 'a',
| 'bb', 'ccc', 'a', 'bb', 'ccc', 'a', 'bb', 'ccc', 'a', 'bb',
| 'ccc', 'a', 'bb', 'ccc', 'a', 'bb', 'ccc', 'a', 'bb', 'ccc'],
| categories=['a', 'bb', 'ccc'], ordered=False, dtype='category')""" # noqa:E501
|
| assert repr(idx) == expected
|
| # truncated
| idx = CategoricalIndex(["a", "bb", "ccc"] * 100)
| expected = """CategoricalIndex(['a', 'bb', 'ccc', 'a', 'bb', 'ccc', 'a', 'bb', 'ccc', 'a',
| ...
| 'ccc', 'a', 'bb', 'ccc', 'a', 'bb', 'ccc', 'a', 'bb', 'ccc'],
| categories=['a', 'bb', 'ccc'], ordered=False, dtype='category', length=300)""" # noqa:E501
|
| assert repr(idx) == expected
|
| # larger categories
| idx = CategoricalIndex(list("abcdefghijklmmo"))
| expected = """CategoricalIndex(['a', 'b', 'c', 'd', 'e', 'f', 'g', 'h', 'i', 'j', 'k', 'l',
| 'm', 'm', 'o'],
| categories=['a', 'b', 'c', 'd', ..., 'k', 'l', 'm', 'o'], ordered=False, dtype='category')""" # noqa:E501
|
| assert repr(idx) == expected
|
| # short
| idx = CategoricalIndex(["あ", "いい", "ううう"])
| expected = """CategoricalIndex(['あ', 'いい', 'ううう'], categories=['あ', 'いい', 'ううう'], ordered=False, dtype='category')""" # noqa:E501
| assert repr(idx) == expected
|
| # multiple lines
| idx = CategoricalIndex(["あ", "いい", "ううう"] * 10)
| expected = """CategoricalIndex(['あ', 'いい', 'ううう', 'あ', 'いい', 'ううう', 'あ', 'いい', 'ううう', 'あ',
| 'いい', 'ううう', 'あ', 'いい', 'ううう', 'あ', 'いい', 'ううう', 'あ', 'いい',
| 'ううう', 'あ', 'いい', 'ううう', 'あ', 'いい', 'ううう', 'あ', 'いい', 'ううう'],
| categories=['あ', 'いい', 'ううう'], ordered=False, dtype='category')""" # noqa:E501
|
| assert repr(idx) == expected
|
| # truncated
| idx = CategoricalIndex(["あ", "いい", "ううう"] * 100)
| expected = """CategoricalIndex(['あ', 'いい', 'ううう', 'あ', 'いい', 'ううう', 'あ', 'いい', 'ううう', 'あ',
| ...
| 'ううう', 'あ', 'いい', 'ううう', 'あ', 'いい', 'ううう', 'あ', 'いい', 'ううう'],
| categories=['あ', 'いい', 'ううう'], ordered=False, dtype='category', length=300)""" # noqa:E501
|
| assert repr(idx) == expected
|
| # larger categories
| idx = CategoricalIndex(list("あいうえおかきくけこさしすせそ"))
| expected = """CategoricalIndex(['あ', 'い', 'う', 'え', 'お', 'か', 'き', 'く', 'け', 'こ', 'さ', 'し',
| 'す', 'せ', 'そ'],
| categories=['あ', 'い', 'う', 'え', ..., 'し', 'す', 'せ', 'そ'], ordered=False, dtype='category')""" # noqa:E501
|
| assert repr(idx) == expected
|
| # Enable Unicode option -----------------------------------------
| with cf.option_context("display.unicode.east_asian_width", True):
| # short
| idx = CategoricalIndex(["あ", "いい", "ううう"])
| expected = """CategoricalIndex(['あ', 'いい', 'ううう'], categories=['あ', 'いい', 'ううう'], ordered=False, dtype='category')""" # noqa:E501
| assert repr(idx) == expected
|
| # multiple lines
| idx = CategoricalIndex(["あ", "いい", "ううう"] * 10)
| expected = """CategoricalIndex(['あ', 'いい', 'ううう', 'あ', 'いい', 'ううう', 'あ', 'いい',
| 'ううう', 'あ', 'いい', 'ううう', 'あ', 'いい', 'ううう',
| 'あ', 'いい', 'ううう', 'あ', 'いい', 'ううう', 'あ', 'いい',
| 'ううう', 'あ', 'いい', 'ううう', 'あ', 'いい', 'ううう'],
| categories=['あ', 'いい', 'ううう'], ordered=False, dtype='category')""" # noqa:E501
|
| assert repr(idx) == expected
|
| # truncated
| idx = CategoricalIndex(["あ", "いい", "ううう"] * 100)
| expected = """CategoricalIndex(['あ', 'いい', 'ううう', 'あ', 'いい', 'ううう', 'あ', 'いい',
| 'ううう', 'あ',
| ...
| 'ううう', 'あ', 'いい', 'ううう', 'あ', 'いい', 'ううう',
| 'あ', 'いい', 'ううう'],
| categories=['あ', 'いい', 'ううう'], ordered=False, dtype='category', length=300)""" # noqa:E501
|
| assert repr(idx) == expected
|
| # larger categories
| idx = CategoricalIndex(list("あいうえおかきくけこさしすせそ"))
| expected = """CategoricalIndex(['あ', 'い', 'う', 'え', 'お', 'か', 'き', 'く', 'け', 'こ',
| 'さ', 'し', 'す', 'せ', 'そ'],
| categories=['あ', 'い', 'う', 'え', ..., 'し', 'す', 'せ', 'そ'], ordered=False, dtype='category')""" # noqa:E501
|
| assert repr(idx) == expected
|
|
