"""
Tests for CategoricalIndex.__repr__ and related methods.
"""

import pandas._config.config as cf

import pandas as pd


class TestCategoricalIndexReprStringCategories:
    def test_string_categorical_index_repr(self):
        # short
        idx = pd.CategoricalIndex(["a", "bb", "ccc"])
        expected = (
            "CategoricalIndex(['a', 'bb', 'ccc'], "
            "categories=['a', 'bb', 'ccc'],\n"
            "                 ordered=False, dtype='category')"
        )
        assert repr(idx) == expected

    def test_categorical_index_repr_multiline(self):
        # multiple lines
        idx = pd.CategoricalIndex(["a", "bb", "ccc"] * 10)
        expected = (
            "CategoricalIndex(['a', 'bb', 'ccc', 'a', 'bb', 'ccc', 'a', "
            "'bb', 'ccc', 'a',\n"
            "                  'bb', 'ccc', 'a', 'bb', 'ccc', 'a', 'bb', "
            "'ccc', 'a', 'bb',\n"
            "                  'ccc', 'a', 'bb', 'ccc', 'a', 'bb', 'ccc', "
            "'a', 'bb', 'ccc'],\n"
            "                 categories=['a', 'bb', 'ccc'], "
            "ordered=False, dtype='category')"
        )
        assert repr(idx) == expected

    def test_categorical_index_repr_truncated(self):
        # truncated
        idx = pd.CategoricalIndex(["a", "bb", "ccc"] * 100)
        expected = (
            "CategoricalIndex(['a', 'bb', 'ccc', 'a', 'bb', 'ccc', 'a', "
            "'bb', 'ccc', 'a',\n"
            "                  ...\n"
            "                  'ccc', 'a', 'bb', 'ccc', 'a', 'bb', 'ccc', "
            "'a', 'bb', 'ccc'],\n"
            "                 categories=['a', 'bb', 'ccc'], "
            "ordered=False, dtype='category',\n"
            "                 length=300)"
        )
        assert repr(idx) == expected

    def test_categorical_index_repr_many_categories(self):
        # larger categories
        idx = pd.CategoricalIndex(list("abcdefghijklmmo"))
        expected = (
            "CategoricalIndex(['a', 'b', 'c', 'd', 'e', 'f', 'g', 'h', "
            "'i', 'j', 'k', 'l',\n"
            "                  'm', 'm', 'o'],\n"
            "                 categories=['a', 'b', 'c', 'd', ..., "
            "'k', 'l', 'm', 'o'],\n"
            "                 ordered=False, dtype='category')"
        )
        assert repr(idx) == expected

    def test_categorical_index_repr_unicode(self):
        # short
        idx = pd.CategoricalIndex(["あ", "いい", "ううう"])
        expected = (
            "CategoricalIndex(['あ', 'いい', 'ううう'], "
            "categories=['あ', 'いい', 'ううう'],\n"
            "                 ordered=False, dtype='category')"
        )
        assert repr(idx) == expected

    def test_categorical_index_repr_unicode_multiline(self):
        # multiple lines
        idx = pd.CategoricalIndex(["あ", "いい", "ううう"] * 10)
        expected = (
            "CategoricalIndex(['あ', 'いい', 'ううう', 'あ', 'いい', 'ううう', "
            "'あ', 'いい', 'ううう', 'あ',\n"
            "                  'いい', 'ううう', 'あ', 'いい', 'ううう', 'あ', "
            "'いい', 'ううう', 'あ', 'いい',\n"
            "                  'ううう', 'あ', 'いい', 'ううう', 'あ', 'いい', "
            "'ううう', 'あ', 'いい', 'ううう'],\n"
            "                 categories=['あ', 'いい', 'ううう'], "
            "ordered=False, dtype='category')"
        )
        assert repr(idx) == expected

    def test_categorical_index_repr_unicode_truncated(self):
        # truncated
        idx = pd.CategoricalIndex(["あ", "いい", "ううう"] * 100)
        expected = (
            "CategoricalIndex(['あ', 'いい', 'ううう', 'あ', 'いい', 'ううう', "
            "'あ', 'いい', 'ううう', 'あ',\n"
            "                  ...\n"
            "                  'ううう', 'あ', 'いい', 'ううう', 'あ', 'いい', "
            "'ううう', 'あ', 'いい', 'ううう'],\n"
            "                 categories=['あ', 'いい', 'ううう'], "
            "ordered=False, dtype='category',\n"
            "                 length=300)"
        )
        assert repr(idx) == expected

    def test_categorical_index_repr_unicode_many_categories(self):
        # larger categories
        idx = pd.CategoricalIndex(list("あいうえおかきくけこさしすせそ"))
        expected = (
            "CategoricalIndex(['あ', 'い', 'う', 'え', 'お', 'か', 'き', "
            "'く', 'け', 'こ', 'さ', 'し',\n"
            "                  'す', 'せ', 'そ'],\n"
            "                 categories=['あ', 'い', 'う', 'え', ..., "
            "'し', 'す', 'せ', 'そ'],\n"
            "                 ordered=False, dtype='category')"
        )
        assert repr(idx) == expected

    def test_categorical_index_repr_east_asian_width(self):
        with cf.option_context("display.unicode.east_asian_width", True):
            # short
            idx = pd.CategoricalIndex(["あ", "いい", "ううう"])
            expected = (
                "CategoricalIndex(['あ', 'いい', 'ううう'], "
                "categories=['あ', 'いい', 'ううう'],\n"
                "                 ordered=False, dtype='category')"
            )
            assert repr(idx) == expected

    def test_categorical_index_repr_east_asian_width_multiline(self):
        with cf.option_context("display.unicode.east_asian_width", True):
            # multiple lines
            idx = pd.CategoricalIndex(["あ", "いい", "ううう"] * 10)
            expected = (
                "CategoricalIndex(['あ', 'いい', 'ううう', 'あ', 'いい', "
                "'ううう', 'あ', 'いい',\n"
                "                  'ううう', 'あ', 'いい', 'ううう', 'あ', "
                "'いい', 'ううう',\n"
                "                  'あ', 'いい', 'ううう', 'あ', 'いい', "
                "'ううう', 'あ', 'いい',\n"
                "                  'ううう', 'あ', 'いい', 'ううう', 'あ', "
                "'いい', 'ううう'],\n"
                "                 categories=['あ', 'いい', 'ううう'], "
                "ordered=False,\n"
                "                 dtype='category')"
            )

            assert repr(idx) == expected

    def test_categorical_index_repr_east_asian_width_truncated(self):
        with cf.option_context("display.unicode.east_asian_width", True):
            # truncated
            idx = pd.CategoricalIndex(["あ", "いい", "ううう"] * 100)
            expected = (
                "CategoricalIndex(['あ', 'いい', 'ううう', 'あ', 'いい', "
                "'ううう', 'あ', 'いい',\n"
                "                  'ううう', 'あ',\n"
                "                  ...\n"
                "                  'ううう', 'あ', 'いい', 'ううう', 'あ', "
                "'いい', 'ううう',\n"
                "                  'あ', 'いい', 'ううう'],\n"
                "                 categories=['あ', 'いい', 'ううう'], "
                "ordered=False,\n"
                "                 dtype='category', length=300)"
            )

            assert repr(idx) == expected

    def test_categorical_index_repr_east_asian_width_many_categories(self):
        with cf.option_context("display.unicode.east_asian_width", True):
            idx = pd.CategoricalIndex(list("あいうえおかきくけこさしすせそ"))
            expected = (
                "CategoricalIndex(['あ', 'い', 'う', 'え', 'お', 'か', 'き', "
                "'く', 'け', 'こ',\n"
                "                  'さ', 'し', 'す', 'せ', 'そ'],\n"
                "                 categories=['あ', 'い', 'う', 'え', ..., "
                "'し', 'す', 'せ', 'そ'],\n"
                "                 ordered=False, dtype='category')"
            )

            assert repr(idx) == expected