517 lines
25 KiB
Python
517 lines
25 KiB
Python
# -*- coding: utf-8 -*-
|
|
|
|
import numpy as np
|
|
|
|
from pandas import (Categorical, Series, CategoricalIndex, date_range,
|
|
period_range, timedelta_range)
|
|
from pandas.compat import u, PY3
|
|
from pandas.core.config import option_context
|
|
from pandas.tests.categorical.common import TestCategorical
|
|
|
|
|
|
class TestCategoricalReprWithFactor(TestCategorical):
|
|
|
|
def test_print(self):
|
|
expected = ["[a, b, b, a, a, c, c, c]",
|
|
"Categories (3, object): [a < b < c]"]
|
|
expected = "\n".join(expected)
|
|
actual = repr(self.factor)
|
|
assert actual == expected
|
|
|
|
|
|
class TestCategoricalRepr(object):
|
|
|
|
def test_big_print(self):
|
|
factor = Categorical([0, 1, 2, 0, 1, 2] * 100, ['a', 'b', 'c'],
|
|
fastpath=True)
|
|
expected = ["[a, b, c, a, b, ..., b, c, a, b, c]", "Length: 600",
|
|
"Categories (3, object): [a, b, c]"]
|
|
expected = "\n".join(expected)
|
|
|
|
actual = repr(factor)
|
|
|
|
assert actual == expected
|
|
|
|
def test_empty_print(self):
|
|
factor = Categorical([], ["a", "b", "c"])
|
|
expected = ("[], Categories (3, object): [a, b, c]")
|
|
# hack because array_repr changed in numpy > 1.6.x
|
|
actual = repr(factor)
|
|
assert actual == expected
|
|
|
|
assert expected == actual
|
|
factor = Categorical([], ["a", "b", "c"], ordered=True)
|
|
expected = ("[], Categories (3, object): [a < b < c]")
|
|
actual = repr(factor)
|
|
assert expected == actual
|
|
|
|
factor = Categorical([], [])
|
|
expected = ("[], Categories (0, object): []")
|
|
assert expected == repr(factor)
|
|
|
|
def test_print_none_width(self):
|
|
# GH10087
|
|
a = Series(Categorical([1, 2, 3, 4]))
|
|
exp = u("0 1\n1 2\n2 3\n3 4\n" +
|
|
"dtype: category\nCategories (4, int64): [1, 2, 3, 4]")
|
|
|
|
with option_context("display.width", None):
|
|
assert exp == repr(a)
|
|
|
|
def test_unicode_print(self):
|
|
if PY3:
|
|
_rep = repr
|
|
else:
|
|
_rep = unicode # noqa
|
|
|
|
c = Categorical(['aaaaa', 'bb', 'cccc'] * 20)
|
|
expected = u"""\
|
|
[aaaaa, bb, cccc, aaaaa, bb, ..., bb, cccc, aaaaa, bb, cccc]
|
|
Length: 60
|
|
Categories (3, object): [aaaaa, bb, cccc]"""
|
|
|
|
assert _rep(c) == expected
|
|
|
|
c = Categorical([u'ああああ', u'いいいいい', u'ううううううう'] * 20)
|
|
expected = u"""\
|
|
[ああああ, いいいいい, ううううううう, ああああ, いいいいい, ..., いいいいい, ううううううう, ああああ, いいいいい, ううううううう]
|
|
Length: 60
|
|
Categories (3, object): [ああああ, いいいいい, ううううううう]""" # noqa
|
|
|
|
assert _rep(c) == expected
|
|
|
|
# unicode option should not affect to Categorical, as it doesn't care
|
|
# the repr width
|
|
with option_context('display.unicode.east_asian_width', True):
|
|
|
|
c = Categorical([u'ああああ', u'いいいいい', u'ううううううう'] * 20)
|
|
expected = u"""[ああああ, いいいいい, ううううううう, ああああ, いいいいい, ..., いいいいい, ううううううう, ああああ, いいいいい, ううううううう]
|
|
Length: 60
|
|
Categories (3, object): [ああああ, いいいいい, ううううううう]""" # noqa
|
|
|
|
assert _rep(c) == expected
|
|
|
|
def test_categorical_repr(self):
|
|
c = Categorical([1, 2, 3])
|
|
exp = """[1, 2, 3]
|
|
Categories (3, int64): [1, 2, 3]"""
|
|
|
|
assert repr(c) == exp
|
|
|
|
c = Categorical([1, 2, 3, 1, 2, 3], categories=[1, 2, 3])
|
|
exp = """[1, 2, 3, 1, 2, 3]
|
|
Categories (3, int64): [1, 2, 3]"""
|
|
|
|
assert repr(c) == exp
|
|
|
|
c = Categorical([1, 2, 3, 4, 5] * 10)
|
|
exp = """[1, 2, 3, 4, 5, ..., 1, 2, 3, 4, 5]
|
|
Length: 50
|
|
Categories (5, int64): [1, 2, 3, 4, 5]"""
|
|
|
|
assert repr(c) == exp
|
|
|
|
c = Categorical(np.arange(20))
|
|
exp = """[0, 1, 2, 3, 4, ..., 15, 16, 17, 18, 19]
|
|
Length: 20
|
|
Categories (20, int64): [0, 1, 2, 3, ..., 16, 17, 18, 19]"""
|
|
|
|
assert repr(c) == exp
|
|
|
|
def test_categorical_repr_ordered(self):
|
|
c = Categorical([1, 2, 3], ordered=True)
|
|
exp = """[1, 2, 3]
|
|
Categories (3, int64): [1 < 2 < 3]"""
|
|
|
|
assert repr(c) == exp
|
|
|
|
c = Categorical([1, 2, 3, 1, 2, 3], categories=[1, 2, 3], ordered=True)
|
|
exp = """[1, 2, 3, 1, 2, 3]
|
|
Categories (3, int64): [1 < 2 < 3]"""
|
|
|
|
assert repr(c) == exp
|
|
|
|
c = Categorical([1, 2, 3, 4, 5] * 10, ordered=True)
|
|
exp = """[1, 2, 3, 4, 5, ..., 1, 2, 3, 4, 5]
|
|
Length: 50
|
|
Categories (5, int64): [1 < 2 < 3 < 4 < 5]"""
|
|
|
|
assert repr(c) == exp
|
|
|
|
c = Categorical(np.arange(20), ordered=True)
|
|
exp = """[0, 1, 2, 3, 4, ..., 15, 16, 17, 18, 19]
|
|
Length: 20
|
|
Categories (20, int64): [0 < 1 < 2 < 3 ... 16 < 17 < 18 < 19]"""
|
|
|
|
assert repr(c) == exp
|
|
|
|
def test_categorical_repr_datetime(self):
|
|
idx = date_range('2011-01-01 09:00', freq='H', periods=5)
|
|
c = Categorical(idx)
|
|
|
|
# TODO(wesm): exceeding 80 characters in the console is not good
|
|
# behavior
|
|
exp = (
|
|
"[2011-01-01 09:00:00, 2011-01-01 10:00:00, 2011-01-01 11:00:00, "
|
|
"2011-01-01 12:00:00, 2011-01-01 13:00:00]\n"
|
|
"Categories (5, datetime64[ns]): [2011-01-01 09:00:00, "
|
|
"2011-01-01 10:00:00, 2011-01-01 11:00:00,\n"
|
|
" 2011-01-01 12:00:00, "
|
|
"2011-01-01 13:00:00]""")
|
|
assert repr(c) == exp
|
|
|
|
c = Categorical(idx.append(idx), categories=idx)
|
|
exp = (
|
|
"[2011-01-01 09:00:00, 2011-01-01 10:00:00, 2011-01-01 11:00:00, "
|
|
"2011-01-01 12:00:00, 2011-01-01 13:00:00, 2011-01-01 09:00:00, "
|
|
"2011-01-01 10:00:00, 2011-01-01 11:00:00, 2011-01-01 12:00:00, "
|
|
"2011-01-01 13:00:00]\n"
|
|
"Categories (5, datetime64[ns]): [2011-01-01 09:00:00, "
|
|
"2011-01-01 10:00:00, 2011-01-01 11:00:00,\n"
|
|
" 2011-01-01 12:00:00, "
|
|
"2011-01-01 13:00:00]")
|
|
|
|
assert repr(c) == exp
|
|
|
|
idx = date_range('2011-01-01 09:00', freq='H', periods=5,
|
|
tz='US/Eastern')
|
|
c = Categorical(idx)
|
|
exp = (
|
|
"[2011-01-01 09:00:00-05:00, 2011-01-01 10:00:00-05:00, "
|
|
"2011-01-01 11:00:00-05:00, 2011-01-01 12:00:00-05:00, "
|
|
"2011-01-01 13:00:00-05:00]\n"
|
|
"Categories (5, datetime64[ns, US/Eastern]): "
|
|
"[2011-01-01 09:00:00-05:00, 2011-01-01 10:00:00-05:00,\n"
|
|
" "
|
|
"2011-01-01 11:00:00-05:00, 2011-01-01 12:00:00-05:00,\n"
|
|
" "
|
|
"2011-01-01 13:00:00-05:00]")
|
|
|
|
assert repr(c) == exp
|
|
|
|
c = Categorical(idx.append(idx), categories=idx)
|
|
exp = (
|
|
"[2011-01-01 09:00:00-05:00, 2011-01-01 10:00:00-05:00, "
|
|
"2011-01-01 11:00:00-05:00, 2011-01-01 12:00:00-05:00, "
|
|
"2011-01-01 13:00:00-05:00, 2011-01-01 09:00:00-05:00, "
|
|
"2011-01-01 10:00:00-05:00, 2011-01-01 11:00:00-05:00, "
|
|
"2011-01-01 12:00:00-05:00, 2011-01-01 13:00:00-05:00]\n"
|
|
"Categories (5, datetime64[ns, US/Eastern]): "
|
|
"[2011-01-01 09:00:00-05:00, 2011-01-01 10:00:00-05:00,\n"
|
|
" "
|
|
"2011-01-01 11:00:00-05:00, 2011-01-01 12:00:00-05:00,\n"
|
|
" "
|
|
"2011-01-01 13:00:00-05:00]")
|
|
|
|
assert repr(c) == exp
|
|
|
|
def test_categorical_repr_datetime_ordered(self):
|
|
idx = date_range('2011-01-01 09:00', freq='H', periods=5)
|
|
c = Categorical(idx, ordered=True)
|
|
exp = """[2011-01-01 09:00:00, 2011-01-01 10:00:00, 2011-01-01 11:00:00, 2011-01-01 12:00:00, 2011-01-01 13:00:00]
|
|
Categories (5, datetime64[ns]): [2011-01-01 09:00:00 < 2011-01-01 10:00:00 < 2011-01-01 11:00:00 <
|
|
2011-01-01 12:00:00 < 2011-01-01 13:00:00]""" # noqa
|
|
|
|
assert repr(c) == exp
|
|
|
|
c = Categorical(idx.append(idx), categories=idx, ordered=True)
|
|
exp = """[2011-01-01 09:00:00, 2011-01-01 10:00:00, 2011-01-01 11:00:00, 2011-01-01 12:00:00, 2011-01-01 13:00:00, 2011-01-01 09:00:00, 2011-01-01 10:00:00, 2011-01-01 11:00:00, 2011-01-01 12:00:00, 2011-01-01 13:00:00]
|
|
Categories (5, datetime64[ns]): [2011-01-01 09:00:00 < 2011-01-01 10:00:00 < 2011-01-01 11:00:00 <
|
|
2011-01-01 12:00:00 < 2011-01-01 13:00:00]""" # noqa
|
|
|
|
assert repr(c) == exp
|
|
|
|
idx = date_range('2011-01-01 09:00', freq='H', periods=5,
|
|
tz='US/Eastern')
|
|
c = Categorical(idx, ordered=True)
|
|
exp = """[2011-01-01 09:00:00-05:00, 2011-01-01 10:00:00-05:00, 2011-01-01 11:00:00-05:00, 2011-01-01 12:00:00-05:00, 2011-01-01 13:00:00-05:00]
|
|
Categories (5, datetime64[ns, US/Eastern]): [2011-01-01 09:00:00-05:00 < 2011-01-01 10:00:00-05:00 <
|
|
2011-01-01 11:00:00-05:00 < 2011-01-01 12:00:00-05:00 <
|
|
2011-01-01 13:00:00-05:00]""" # noqa
|
|
|
|
assert repr(c) == exp
|
|
|
|
c = Categorical(idx.append(idx), categories=idx, ordered=True)
|
|
exp = """[2011-01-01 09:00:00-05:00, 2011-01-01 10:00:00-05:00, 2011-01-01 11:00:00-05:00, 2011-01-01 12:00:00-05:00, 2011-01-01 13:00:00-05:00, 2011-01-01 09:00:00-05:00, 2011-01-01 10:00:00-05:00, 2011-01-01 11:00:00-05:00, 2011-01-01 12:00:00-05:00, 2011-01-01 13:00:00-05:00]
|
|
Categories (5, datetime64[ns, US/Eastern]): [2011-01-01 09:00:00-05:00 < 2011-01-01 10:00:00-05:00 <
|
|
2011-01-01 11:00:00-05:00 < 2011-01-01 12:00:00-05:00 <
|
|
2011-01-01 13:00:00-05:00]""" # noqa
|
|
|
|
assert repr(c) == exp
|
|
|
|
def test_categorical_repr_period(self):
|
|
idx = period_range('2011-01-01 09:00', freq='H', periods=5)
|
|
c = Categorical(idx)
|
|
exp = """[2011-01-01 09:00, 2011-01-01 10:00, 2011-01-01 11:00, 2011-01-01 12:00, 2011-01-01 13:00]
|
|
Categories (5, period[H]): [2011-01-01 09:00, 2011-01-01 10:00, 2011-01-01 11:00, 2011-01-01 12:00,
|
|
2011-01-01 13:00]""" # noqa
|
|
|
|
assert repr(c) == exp
|
|
|
|
c = Categorical(idx.append(idx), categories=idx)
|
|
exp = """[2011-01-01 09:00, 2011-01-01 10:00, 2011-01-01 11:00, 2011-01-01 12:00, 2011-01-01 13:00, 2011-01-01 09:00, 2011-01-01 10:00, 2011-01-01 11:00, 2011-01-01 12:00, 2011-01-01 13:00]
|
|
Categories (5, period[H]): [2011-01-01 09:00, 2011-01-01 10:00, 2011-01-01 11:00, 2011-01-01 12:00,
|
|
2011-01-01 13:00]""" # noqa
|
|
|
|
assert repr(c) == exp
|
|
|
|
idx = period_range('2011-01', freq='M', periods=5)
|
|
c = Categorical(idx)
|
|
exp = """[2011-01, 2011-02, 2011-03, 2011-04, 2011-05]
|
|
Categories (5, period[M]): [2011-01, 2011-02, 2011-03, 2011-04, 2011-05]"""
|
|
|
|
assert repr(c) == exp
|
|
|
|
c = Categorical(idx.append(idx), categories=idx)
|
|
exp = """[2011-01, 2011-02, 2011-03, 2011-04, 2011-05, 2011-01, 2011-02, 2011-03, 2011-04, 2011-05]
|
|
Categories (5, period[M]): [2011-01, 2011-02, 2011-03, 2011-04, 2011-05]""" # noqa
|
|
|
|
assert repr(c) == exp
|
|
|
|
def test_categorical_repr_period_ordered(self):
|
|
idx = period_range('2011-01-01 09:00', freq='H', periods=5)
|
|
c = Categorical(idx, ordered=True)
|
|
exp = """[2011-01-01 09:00, 2011-01-01 10:00, 2011-01-01 11:00, 2011-01-01 12:00, 2011-01-01 13:00]
|
|
Categories (5, period[H]): [2011-01-01 09:00 < 2011-01-01 10:00 < 2011-01-01 11:00 < 2011-01-01 12:00 <
|
|
2011-01-01 13:00]""" # noqa
|
|
|
|
assert repr(c) == exp
|
|
|
|
c = Categorical(idx.append(idx), categories=idx, ordered=True)
|
|
exp = """[2011-01-01 09:00, 2011-01-01 10:00, 2011-01-01 11:00, 2011-01-01 12:00, 2011-01-01 13:00, 2011-01-01 09:00, 2011-01-01 10:00, 2011-01-01 11:00, 2011-01-01 12:00, 2011-01-01 13:00]
|
|
Categories (5, period[H]): [2011-01-01 09:00 < 2011-01-01 10:00 < 2011-01-01 11:00 < 2011-01-01 12:00 <
|
|
2011-01-01 13:00]""" # noqa
|
|
|
|
assert repr(c) == exp
|
|
|
|
idx = period_range('2011-01', freq='M', periods=5)
|
|
c = Categorical(idx, ordered=True)
|
|
exp = """[2011-01, 2011-02, 2011-03, 2011-04, 2011-05]
|
|
Categories (5, period[M]): [2011-01 < 2011-02 < 2011-03 < 2011-04 < 2011-05]"""
|
|
|
|
assert repr(c) == exp
|
|
|
|
c = Categorical(idx.append(idx), categories=idx, ordered=True)
|
|
exp = """[2011-01, 2011-02, 2011-03, 2011-04, 2011-05, 2011-01, 2011-02, 2011-03, 2011-04, 2011-05]
|
|
Categories (5, period[M]): [2011-01 < 2011-02 < 2011-03 < 2011-04 < 2011-05]""" # noqa
|
|
|
|
assert repr(c) == exp
|
|
|
|
def test_categorical_repr_timedelta(self):
|
|
idx = timedelta_range('1 days', periods=5)
|
|
c = Categorical(idx)
|
|
exp = """[1 days, 2 days, 3 days, 4 days, 5 days]
|
|
Categories (5, timedelta64[ns]): [1 days, 2 days, 3 days, 4 days, 5 days]"""
|
|
|
|
assert repr(c) == exp
|
|
|
|
c = Categorical(idx.append(idx), categories=idx)
|
|
exp = """[1 days, 2 days, 3 days, 4 days, 5 days, 1 days, 2 days, 3 days, 4 days, 5 days]
|
|
Categories (5, timedelta64[ns]): [1 days, 2 days, 3 days, 4 days, 5 days]""" # noqa
|
|
|
|
assert repr(c) == exp
|
|
|
|
idx = timedelta_range('1 hours', periods=20)
|
|
c = Categorical(idx)
|
|
exp = """[0 days 01:00:00, 1 days 01:00:00, 2 days 01:00:00, 3 days 01:00:00, 4 days 01:00:00, ..., 15 days 01:00:00, 16 days 01:00:00, 17 days 01:00:00, 18 days 01:00:00, 19 days 01:00:00]
|
|
Length: 20
|
|
Categories (20, timedelta64[ns]): [0 days 01:00:00, 1 days 01:00:00, 2 days 01:00:00,
|
|
3 days 01:00:00, ..., 16 days 01:00:00, 17 days 01:00:00,
|
|
18 days 01:00:00, 19 days 01:00:00]""" # noqa
|
|
|
|
assert repr(c) == exp
|
|
|
|
c = Categorical(idx.append(idx), categories=idx)
|
|
exp = """[0 days 01:00:00, 1 days 01:00:00, 2 days 01:00:00, 3 days 01:00:00, 4 days 01:00:00, ..., 15 days 01:00:00, 16 days 01:00:00, 17 days 01:00:00, 18 days 01:00:00, 19 days 01:00:00]
|
|
Length: 40
|
|
Categories (20, timedelta64[ns]): [0 days 01:00:00, 1 days 01:00:00, 2 days 01:00:00,
|
|
3 days 01:00:00, ..., 16 days 01:00:00, 17 days 01:00:00,
|
|
18 days 01:00:00, 19 days 01:00:00]""" # noqa
|
|
|
|
assert repr(c) == exp
|
|
|
|
def test_categorical_repr_timedelta_ordered(self):
|
|
idx = timedelta_range('1 days', periods=5)
|
|
c = Categorical(idx, ordered=True)
|
|
exp = """[1 days, 2 days, 3 days, 4 days, 5 days]
|
|
Categories (5, timedelta64[ns]): [1 days < 2 days < 3 days < 4 days < 5 days]""" # noqa
|
|
|
|
assert repr(c) == exp
|
|
|
|
c = Categorical(idx.append(idx), categories=idx, ordered=True)
|
|
exp = """[1 days, 2 days, 3 days, 4 days, 5 days, 1 days, 2 days, 3 days, 4 days, 5 days]
|
|
Categories (5, timedelta64[ns]): [1 days < 2 days < 3 days < 4 days < 5 days]""" # noqa
|
|
|
|
assert repr(c) == exp
|
|
|
|
idx = timedelta_range('1 hours', periods=20)
|
|
c = Categorical(idx, ordered=True)
|
|
exp = """[0 days 01:00:00, 1 days 01:00:00, 2 days 01:00:00, 3 days 01:00:00, 4 days 01:00:00, ..., 15 days 01:00:00, 16 days 01:00:00, 17 days 01:00:00, 18 days 01:00:00, 19 days 01:00:00]
|
|
Length: 20
|
|
Categories (20, timedelta64[ns]): [0 days 01:00:00 < 1 days 01:00:00 < 2 days 01:00:00 <
|
|
3 days 01:00:00 ... 16 days 01:00:00 < 17 days 01:00:00 <
|
|
18 days 01:00:00 < 19 days 01:00:00]""" # noqa
|
|
|
|
assert repr(c) == exp
|
|
|
|
c = Categorical(idx.append(idx), categories=idx, ordered=True)
|
|
exp = """[0 days 01:00:00, 1 days 01:00:00, 2 days 01:00:00, 3 days 01:00:00, 4 days 01:00:00, ..., 15 days 01:00:00, 16 days 01:00:00, 17 days 01:00:00, 18 days 01:00:00, 19 days 01:00:00]
|
|
Length: 40
|
|
Categories (20, timedelta64[ns]): [0 days 01:00:00 < 1 days 01:00:00 < 2 days 01:00:00 <
|
|
3 days 01:00:00 ... 16 days 01:00:00 < 17 days 01:00:00 <
|
|
18 days 01:00:00 < 19 days 01:00:00]""" # noqa
|
|
|
|
assert repr(c) == exp
|
|
|
|
def test_categorical_index_repr(self):
|
|
idx = CategoricalIndex(Categorical([1, 2, 3]))
|
|
exp = """CategoricalIndex([1, 2, 3], categories=[1, 2, 3], ordered=False, dtype='category')""" # noqa
|
|
assert repr(idx) == exp
|
|
|
|
i = CategoricalIndex(Categorical(np.arange(10)))
|
|
exp = """CategoricalIndex([0, 1, 2, 3, 4, 5, 6, 7, 8, 9], categories=[0, 1, 2, 3, 4, 5, 6, 7, ...], ordered=False, dtype='category')""" # noqa
|
|
assert repr(i) == exp
|
|
|
|
def test_categorical_index_repr_ordered(self):
|
|
i = CategoricalIndex(Categorical([1, 2, 3], ordered=True))
|
|
exp = """CategoricalIndex([1, 2, 3], categories=[1, 2, 3], ordered=True, dtype='category')""" # noqa
|
|
assert repr(i) == exp
|
|
|
|
i = CategoricalIndex(Categorical(np.arange(10), ordered=True))
|
|
exp = """CategoricalIndex([0, 1, 2, 3, 4, 5, 6, 7, 8, 9], categories=[0, 1, 2, 3, 4, 5, 6, 7, ...], ordered=True, dtype='category')""" # noqa
|
|
assert repr(i) == exp
|
|
|
|
def test_categorical_index_repr_datetime(self):
|
|
idx = date_range('2011-01-01 09:00', freq='H', periods=5)
|
|
i = CategoricalIndex(Categorical(idx))
|
|
exp = """CategoricalIndex(['2011-01-01 09:00:00', '2011-01-01 10:00:00',
|
|
'2011-01-01 11:00:00', '2011-01-01 12:00:00',
|
|
'2011-01-01 13:00:00'],
|
|
categories=[2011-01-01 09:00:00, 2011-01-01 10:00:00, 2011-01-01 11:00:00, 2011-01-01 12:00:00, 2011-01-01 13:00:00], ordered=False, dtype='category')""" # noqa
|
|
|
|
assert repr(i) == exp
|
|
|
|
idx = date_range('2011-01-01 09:00', freq='H', periods=5,
|
|
tz='US/Eastern')
|
|
i = CategoricalIndex(Categorical(idx))
|
|
exp = """CategoricalIndex(['2011-01-01 09:00:00-05:00', '2011-01-01 10:00:00-05:00',
|
|
'2011-01-01 11:00:00-05:00', '2011-01-01 12:00:00-05:00',
|
|
'2011-01-01 13:00:00-05:00'],
|
|
categories=[2011-01-01 09:00:00-05:00, 2011-01-01 10:00:00-05:00, 2011-01-01 11:00:00-05:00, 2011-01-01 12:00:00-05:00, 2011-01-01 13:00:00-05:00], ordered=False, dtype='category')""" # noqa
|
|
|
|
assert repr(i) == exp
|
|
|
|
def test_categorical_index_repr_datetime_ordered(self):
|
|
idx = date_range('2011-01-01 09:00', freq='H', periods=5)
|
|
i = CategoricalIndex(Categorical(idx, ordered=True))
|
|
exp = """CategoricalIndex(['2011-01-01 09:00:00', '2011-01-01 10:00:00',
|
|
'2011-01-01 11:00:00', '2011-01-01 12:00:00',
|
|
'2011-01-01 13:00:00'],
|
|
categories=[2011-01-01 09:00:00, 2011-01-01 10:00:00, 2011-01-01 11:00:00, 2011-01-01 12:00:00, 2011-01-01 13:00:00], ordered=True, dtype='category')""" # noqa
|
|
|
|
assert repr(i) == exp
|
|
|
|
idx = date_range('2011-01-01 09:00', freq='H', periods=5,
|
|
tz='US/Eastern')
|
|
i = CategoricalIndex(Categorical(idx, ordered=True))
|
|
exp = """CategoricalIndex(['2011-01-01 09:00:00-05:00', '2011-01-01 10:00:00-05:00',
|
|
'2011-01-01 11:00:00-05:00', '2011-01-01 12:00:00-05:00',
|
|
'2011-01-01 13:00:00-05:00'],
|
|
categories=[2011-01-01 09:00:00-05:00, 2011-01-01 10:00:00-05:00, 2011-01-01 11:00:00-05:00, 2011-01-01 12:00:00-05:00, 2011-01-01 13:00:00-05:00], ordered=True, dtype='category')""" # noqa
|
|
|
|
assert repr(i) == exp
|
|
|
|
i = CategoricalIndex(Categorical(idx.append(idx), ordered=True))
|
|
exp = """CategoricalIndex(['2011-01-01 09:00:00-05:00', '2011-01-01 10:00:00-05:00',
|
|
'2011-01-01 11:00:00-05:00', '2011-01-01 12:00:00-05:00',
|
|
'2011-01-01 13:00:00-05:00', '2011-01-01 09:00:00-05:00',
|
|
'2011-01-01 10:00:00-05:00', '2011-01-01 11:00:00-05:00',
|
|
'2011-01-01 12:00:00-05:00', '2011-01-01 13:00:00-05:00'],
|
|
categories=[2011-01-01 09:00:00-05:00, 2011-01-01 10:00:00-05:00, 2011-01-01 11:00:00-05:00, 2011-01-01 12:00:00-05:00, 2011-01-01 13:00:00-05:00], ordered=True, dtype='category')""" # noqa
|
|
|
|
assert repr(i) == exp
|
|
|
|
def test_categorical_index_repr_period(self):
|
|
# test all length
|
|
idx = period_range('2011-01-01 09:00', freq='H', periods=1)
|
|
i = CategoricalIndex(Categorical(idx))
|
|
exp = """CategoricalIndex(['2011-01-01 09:00'], categories=[2011-01-01 09:00], ordered=False, dtype='category')""" # noqa
|
|
assert repr(i) == exp
|
|
|
|
idx = period_range('2011-01-01 09:00', freq='H', periods=2)
|
|
i = CategoricalIndex(Categorical(idx))
|
|
exp = """CategoricalIndex(['2011-01-01 09:00', '2011-01-01 10:00'], categories=[2011-01-01 09:00, 2011-01-01 10:00], ordered=False, dtype='category')""" # noqa
|
|
assert repr(i) == exp
|
|
|
|
idx = period_range('2011-01-01 09:00', freq='H', periods=3)
|
|
i = CategoricalIndex(Categorical(idx))
|
|
exp = """CategoricalIndex(['2011-01-01 09:00', '2011-01-01 10:00', '2011-01-01 11:00'], categories=[2011-01-01 09:00, 2011-01-01 10:00, 2011-01-01 11:00], ordered=False, dtype='category')""" # noqa
|
|
assert repr(i) == exp
|
|
|
|
idx = period_range('2011-01-01 09:00', freq='H', periods=5)
|
|
i = CategoricalIndex(Categorical(idx))
|
|
exp = """CategoricalIndex(['2011-01-01 09:00', '2011-01-01 10:00', '2011-01-01 11:00',
|
|
'2011-01-01 12:00', '2011-01-01 13:00'],
|
|
categories=[2011-01-01 09:00, 2011-01-01 10:00, 2011-01-01 11:00, 2011-01-01 12:00, 2011-01-01 13:00], ordered=False, dtype='category')""" # noqa
|
|
|
|
assert repr(i) == exp
|
|
|
|
i = CategoricalIndex(Categorical(idx.append(idx)))
|
|
exp = """CategoricalIndex(['2011-01-01 09:00', '2011-01-01 10:00', '2011-01-01 11:00',
|
|
'2011-01-01 12:00', '2011-01-01 13:00', '2011-01-01 09:00',
|
|
'2011-01-01 10:00', '2011-01-01 11:00', '2011-01-01 12:00',
|
|
'2011-01-01 13:00'],
|
|
categories=[2011-01-01 09:00, 2011-01-01 10:00, 2011-01-01 11:00, 2011-01-01 12:00, 2011-01-01 13:00], ordered=False, dtype='category')""" # noqa
|
|
|
|
assert repr(i) == exp
|
|
|
|
idx = period_range('2011-01', freq='M', periods=5)
|
|
i = CategoricalIndex(Categorical(idx))
|
|
exp = """CategoricalIndex(['2011-01', '2011-02', '2011-03', '2011-04', '2011-05'], categories=[2011-01, 2011-02, 2011-03, 2011-04, 2011-05], ordered=False, dtype='category')""" # noqa
|
|
assert repr(i) == exp
|
|
|
|
def test_categorical_index_repr_period_ordered(self):
|
|
idx = period_range('2011-01-01 09:00', freq='H', periods=5)
|
|
i = CategoricalIndex(Categorical(idx, ordered=True))
|
|
exp = """CategoricalIndex(['2011-01-01 09:00', '2011-01-01 10:00', '2011-01-01 11:00',
|
|
'2011-01-01 12:00', '2011-01-01 13:00'],
|
|
categories=[2011-01-01 09:00, 2011-01-01 10:00, 2011-01-01 11:00, 2011-01-01 12:00, 2011-01-01 13:00], ordered=True, dtype='category')""" # noqa
|
|
|
|
assert repr(i) == exp
|
|
|
|
idx = period_range('2011-01', freq='M', periods=5)
|
|
i = CategoricalIndex(Categorical(idx, ordered=True))
|
|
exp = """CategoricalIndex(['2011-01', '2011-02', '2011-03', '2011-04', '2011-05'], categories=[2011-01, 2011-02, 2011-03, 2011-04, 2011-05], ordered=True, dtype='category')""" # noqa
|
|
assert repr(i) == exp
|
|
|
|
def test_categorical_index_repr_timedelta(self):
|
|
idx = timedelta_range('1 days', periods=5)
|
|
i = CategoricalIndex(Categorical(idx))
|
|
exp = """CategoricalIndex(['1 days', '2 days', '3 days', '4 days', '5 days'], categories=[1 days 00:00:00, 2 days 00:00:00, 3 days 00:00:00, 4 days 00:00:00, 5 days 00:00:00], ordered=False, dtype='category')""" # noqa
|
|
assert repr(i) == exp
|
|
|
|
idx = timedelta_range('1 hours', periods=10)
|
|
i = CategoricalIndex(Categorical(idx))
|
|
exp = """CategoricalIndex(['0 days 01:00:00', '1 days 01:00:00', '2 days 01:00:00',
|
|
'3 days 01:00:00', '4 days 01:00:00', '5 days 01:00:00',
|
|
'6 days 01:00:00', '7 days 01:00:00', '8 days 01:00:00',
|
|
'9 days 01:00:00'],
|
|
categories=[0 days 01:00:00, 1 days 01:00:00, 2 days 01:00:00, 3 days 01:00:00, 4 days 01:00:00, 5 days 01:00:00, 6 days 01:00:00, 7 days 01:00:00, ...], ordered=False, dtype='category')""" # noqa
|
|
|
|
assert repr(i) == exp
|
|
|
|
def test_categorical_index_repr_timedelta_ordered(self):
|
|
idx = timedelta_range('1 days', periods=5)
|
|
i = CategoricalIndex(Categorical(idx, ordered=True))
|
|
exp = """CategoricalIndex(['1 days', '2 days', '3 days', '4 days', '5 days'], categories=[1 days 00:00:00, 2 days 00:00:00, 3 days 00:00:00, 4 days 00:00:00, 5 days 00:00:00], ordered=True, dtype='category')""" # noqa
|
|
assert repr(i) == exp
|
|
|
|
idx = timedelta_range('1 hours', periods=10)
|
|
i = CategoricalIndex(Categorical(idx, ordered=True))
|
|
exp = """CategoricalIndex(['0 days 01:00:00', '1 days 01:00:00', '2 days 01:00:00',
|
|
'3 days 01:00:00', '4 days 01:00:00', '5 days 01:00:00',
|
|
'6 days 01:00:00', '7 days 01:00:00', '8 days 01:00:00',
|
|
'9 days 01:00:00'],
|
|
categories=[0 days 01:00:00, 1 days 01:00:00, 2 days 01:00:00, 3 days 01:00:00, 4 days 01:00:00, 5 days 01:00:00, 6 days 01:00:00, 7 days 01:00:00, ...], ordered=True, dtype='category')""" # noqa
|
|
|
|
assert repr(i) == exp
|