1 files changed, 185 insertions, 0 deletions
diff --git a/tests/test_color.py b/tests/test_color.py
new file mode 100644
index 0000000..d5c7170
--- /dev/null
+++ b/tests/test_color.py
@@ -0,0 +1,185 @@
+"""Test objects in module."""
+
+import sys
+from functools import partial
+
+import pytest
+
+from colorclass.color import Color
+from tests.conftest import assert_both_values, get_instance
+
+
+def test_colorize_methods():
+    """Test colorize convenience methods."""
+    assert Color.black('TEST').value_colors == '\033[30mTEST\033[39m'
+    assert Color.bgblack('TEST').value_colors == '\033[40mTEST\033[49m'
+    assert Color.red('TEST').value_colors == '\033[31mTEST\033[39m'
+    assert Color.bgred('TEST').value_colors == '\033[41mTEST\033[49m'
+    assert Color.green('TEST').value_colors == '\033[32mTEST\033[39m'
+    assert Color.bggreen('TEST').value_colors == '\033[42mTEST\033[49m'
+    assert Color.yellow('TEST').value_colors == '\033[33mTEST\033[39m'
+    assert Color.bgyellow('TEST').value_colors == '\033[43mTEST\033[49m'
+    assert Color.blue('TEST').value_colors == '\033[34mTEST\033[39m'
+    assert Color.bgblue('TEST').value_colors == '\033[44mTEST\033[49m'
+    assert Color.magenta('TEST').value_colors == '\033[35mTEST\033[39m'
+    assert Color.bgmagenta('TEST').value_colors == '\033[45mTEST\033[49m'
+    assert Color.cyan('TEST').value_colors == '\033[36mTEST\033[39m'
+    assert Color.bgcyan('TEST').value_colors == '\033[46mTEST\033[49m'
+    assert Color.white('TEST').value_colors == '\033[37mTEST\033[39m'
+    assert Color.bgwhite('TEST').value_colors == '\033[47mTEST\033[49m'
+
+    assert Color.black('this is a test.', auto=True) == Color('{autoblack}this is a test.{/autoblack}')
+    assert Color.black('this is a test.') == Color('{black}this is a test.{/black}')
+    assert Color.bgblack('this is a test.', auto=True) == Color('{autobgblack}this is a test.{/autobgblack}')
+    assert Color.bgblack('this is a test.') == Color('{bgblack}this is a test.{/bgblack}')
+    assert Color.red('this is a test.', auto=True) == Color('{autored}this is a test.{/autored}')
+    assert Color.red('this is a test.') == Color('{red}this is a test.{/red}')
+    assert Color.bgred('this is a test.', auto=True) == Color('{autobgred}this is a test.{/autobgred}')
+    assert Color.bgred('this is a test.') == Color('{bgred}this is a test.{/bgred}')
+    assert Color.green('this is a test.', auto=True) == Color('{autogreen}this is a test.{/autogreen}')
+    assert Color.green('this is a test.') == Color('{green}this is a test.{/green}')
+    assert Color.bggreen('this is a test.', auto=True) == Color('{autobggreen}this is a test.{/autobggreen}')
+    assert Color.bggreen('this is a test.') == Color('{bggreen}this is a test.{/bggreen}')
+    assert Color.yellow('this is a test.', auto=True) == Color('{autoyellow}this is a test.{/autoyellow}')
+    assert Color.yellow('this is a test.') == Color('{yellow}this is a test.{/yellow}')
+    assert Color.bgyellow('this is a test.', auto=True) == Color('{autobgyellow}this is a test.{/autobgyellow}')
+    assert Color.bgyellow('this is a test.') == Color('{bgyellow}this is a test.{/bgyellow}')
+    assert Color.blue('this is a test.', auto=True) == Color('{autoblue}this is a test.{/autoblue}')
+    assert Color.blue('this is a test.') == Color('{blue}this is a test.{/blue}')
+    assert Color.bgblue('this is a test.', auto=True) == Color('{autobgblue}this is a test.{/autobgblue}')
+    assert Color.bgblue('this is a test.') == Color('{bgblue}this is a test.{/bgblue}')
+    assert Color.magenta('this is a test.', auto=True) == Color('{automagenta}this is a test.{/automagenta}')
+    assert Color.magenta('this is a test.') == Color('{magenta}this is a test.{/magenta}')
+    assert Color.bgmagenta('this is a test.', auto=True) == Color('{autobgmagenta}this is a test.{/autobgmagenta}')
+    assert Color.bgmagenta('this is a test.') == Color('{bgmagenta}this is a test.{/bgmagenta}')
+    assert Color.cyan('this is a test.', auto=True) == Color('{autocyan}this is a test.{/autocyan}')
+    assert Color.cyan('this is a test.') == Color('{cyan}this is a test.{/cyan}')
+    assert Color.bgcyan('this is a test.', auto=True) == Color('{autobgcyan}this is a test.{/autobgcyan}')
+    assert Color.bgcyan('this is a test.') == Color('{bgcyan}this is a test.{/bgcyan}')
+    assert Color.white('this is a test.', auto=True) == Color('{autowhite}this is a test.{/autowhite}')
+    assert Color.white('this is a test.') == Color('{white}this is a test.{/white}')
+    assert Color.bgwhite('this is a test.', auto=True) == Color('{autobgwhite}this is a test.{/autobgwhite}')
+    assert Color.bgwhite('this is a test.') == Color('{bgwhite}this is a test.{/bgwhite}')
+
+
+@pytest.mark.parametrize('kind', ['str', 'Color plain', 'Color color'])
+def test_chaining(kind):
+    """Test chaining Color instances.
+
+    :param str kind: Type of string to test.
+    """
+    assert_both = partial(assert_both_values, kind=kind)
+
+    # Test string.
+    instance = get_instance(kind, 'TEST')
+    for color in ('green', 'blue', 'yellow'):
+        instance = get_instance(kind, instance, color)
+    assert_both(instance, 'TEST', '\033[31mTEST\033[39m')
+
+    # Test empty.
+    instance = get_instance(kind)
+    for color in ('red', 'green', 'blue', 'yellow'):
+        instance = get_instance(kind, instance, color)
+    assert_both(instance, '', '\033[39m')
+
+    # Test complicated.
+    instance = 'TEST'
+    for color in ('black', 'bgred', 'green', 'bgyellow', 'blue', 'bgmagenta', 'cyan', 'bgwhite'):
+        instance = get_instance(kind, instance, color=color)
+    assert_both(instance, 'TEST', '\033[30;41mTEST\033[39;49m')
+
+    # Test format and length.
+    instance = get_instance(kind, '{0}').format(get_instance(kind, 'TEST'))
+    assert_both(instance, 'TEST', '\033[31mTEST\033[39m')
+    assert len(instance) == 4
+    instance = get_instance(kind, '{0}').format(instance)
+    assert_both(instance, 'TEST', '\033[31mTEST\033[39m')
+    assert len(instance) == 4
+    instance = get_instance(kind, '{0}').format(instance)
+    assert_both(instance, 'TEST', '\033[31mTEST\033[39m')
+    assert len(instance) == 4
+
+
+@pytest.mark.parametrize('kind', ['str', 'Color plain', 'Color color'])
+def test_empty(kind):
+    """Test with empty string.
+
+    :param str kind: Type of string to test.
+    """
+    instance = get_instance(kind, u'')
+    assert_both = partial(assert_both_values, kind=kind)
+
+    assert len(instance) == 0
+    assert_both(instance * 2, '', '\033[39m')
+    assert_both(instance + instance, '', '\033[39m')
+    with pytest.raises(IndexError):
+        assert instance[0]
+    assert not [i for i in instance]
+    assert not list(instance)
+
+    assert instance.encode('utf-8') == instance.encode('utf-8')
+    assert instance.encode('utf-8').decode('utf-8') == instance
+    assert_both(instance.encode('utf-8').decode('utf-8'), '', '\033[39m')
+    assert_both(instance.__class__.encode(instance, 'utf-8').decode('utf-8'), '', '\033[39m')
+    assert len(instance.encode('utf-8').decode('utf-8')) == 0
+    assert_both(instance.format(value=''), '', '\033[39m')
+
+    assert_both(instance.capitalize(), '', '\033[39m')
+    assert_both(instance.center(5), '     ', '\033[39m     ')
+    assert instance.count('') == 1
+    assert instance.count('t') == 0
+    assert instance.endswith('') is True
+    assert instance.endswith('me') is False
+    assert instance.find('') == 0
+    assert instance.find('t') == -1
+
+    assert instance.index('') == 0
+    with pytest.raises(ValueError):
+        assert instance.index('t')
+    assert instance.isalnum() is False
+    assert instance.isalpha() is False
+    if sys.version_info[0] != 2:
+        assert instance.isdecimal() is False
+    assert instance.isdigit() is False
+    if sys.version_info[0] != 2:
+        assert instance.isnumeric() is False
+    assert instance.isspace() is False
+    assert instance.istitle() is False
+    assert instance.isupper() is False
+
+    assert_both(instance.join(['A', 'B']), 'AB', 'A\033[39mB')
+    assert_both(instance.ljust(5), '     ', '\033[39m     ')
+    assert instance.rfind('') == 0
+    assert instance.rfind('t') == -1
+    assert instance.rindex('') == 0
+    with pytest.raises(ValueError):
+        assert instance.rindex('t')
+    assert_both(instance.rjust(5), '     ', '\033[39m     ')
+    if kind in ('str', 'Color plain'):
+        assert instance.splitlines() == list()
+    else:
+        assert instance.splitlines() == ['\033[39m']
+    assert instance.startswith('') is True
+    assert instance.startswith('T') is False
+    assert_both(instance.swapcase(), '', '\033[39m')
+
+    assert_both(instance.title(), '', '\033[39m')
+    assert_both(instance.translate({ord('t'): u'1', ord('e'): u'2', ord('s'): u'3'}), '', '\033[39m')
+    assert_both(instance.upper(), '', '\033[39m')
+    assert_both(instance.zfill(0), '', '')
+    assert_both(instance.zfill(1), '0', '0')
+
+
+def test_keep_tags():
+    """Test keep_tags keyword arg."""
+    assert_both = partial(assert_both_values, kind='Color color')
+
+    instance = Color('{red}Test{/red}', keep_tags=True)
+    assert_both(instance, '{red}Test{/red}', '{red}Test{/red}')
+    assert_both(instance.upper(), '{RED}TEST{/RED}', '{RED}TEST{/RED}')
+    assert len(instance) == 15
+
+    instance = Color('{red}\033[41mTest\033[49m{/red}', keep_tags=True)
+    assert_both(instance, '{red}Test{/red}', '{red}\033[41mTest\033[49m{/red}')
+    assert_both(instance.upper(), '{RED}TEST{/RED}', '{RED}\033[41mTEST\033[49m{/RED}')
+    assert len(instance) == 15