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
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
|
# -*- encoding: utf-8 -*-
"""
Tests for colors module
"""
from colors import *
from pytest import raises
import pytest
import sys
_PY2 = sys.version_info[0] == 2
def test_nocolors():
"""We should have the original text with no parameters"""
assert color("RED") == "RED"
def test_redcolor():
"""Test if we get a correct red"""
assert color("RED", fg="red") == '\x1b[31mRED\x1b[0m'
def test_default_color():
assert color('test', fg='default') == '\x1b[39mtest\x1b[0m'
assert color('test', bg='default') == '\x1b[49mtest\x1b[0m'
assert color('test', fg='default', bg='default') == \
'\x1b[39;49mtest\x1b[0m'
def test_error_on_bad_color_string():
"""Test if we have an exception with a bad string color"""
with raises(ValueError):
color("RED", "bozo")
def test_integer_color():
"""Test if we can use a integer as a color"""
assert color("RED", 1) == '\x1b[38;5;1mRED\x1b[0m'
def test_error_on_bad_color_int():
"""Test if we have an exception with a bad color number"""
with raises(ValueError):
color("RED", 911)
def test_tuple_color():
orange = (255, 165, 0)
assert color('ORANGE', fg=orange) == '\x1b[38;2;255;165;0mORANGE\x1b[0m'
def test_css_color():
assert color('ORANGE', fg='orange') == '\x1b[38;2;255;165;0mORANGE\x1b[0m'
def test_background_color():
"""Test the background color with string"""
assert color("RED", bg="red") == '\x1b[41mRED\x1b[0m'
assert color('PURPLE', bg='purple') == '\x1b[48;2;128;0;128mPURPLE\x1b[0m'
def test_mixed_color():
assert color('PINK/GRAY', fg='pink', bg='gray') == \
'\x1b[38;2;255;192;203;48;2;128;128;128mPINK/GRAY\x1b[0m'
assert color('GRAY/PINK', fg='gray', bg='pink') == \
'\x1b[38;2;128;128;128;48;2;255;192;203mGRAY/PINK\x1b[0m'
def test_style_color():
"""Test the style with a string"""
assert color('BOLD', style='bold') == '\x1b[1mBOLD\x1b[0m'
def test_error_setting_style_color():
"""Test if we have an exception setting the style"""
with raises(ValueError):
color("BOLD", style="MAY")
def test_error_on_bad_color_notint():
"""Test if we have an exception with a bad color number"""
with raises(ValueError):
color("RED", 911.11)
def test_error_on_bad_style():
"""Test if we have an exception with a bad color number"""
with raises(ValueError):
color("RED", bg="cursivas")
def test_remove_color():
"""We can get the original message without the colors"""
assert strip_color(color("RED", "red")) == "RED"
assert strip_color(color("text", "red", "green", "bold+underline")) == "text"
def test_remove_color_extra():
"""Test other sequences that color would not add, but that are seen in the wild."""
assert strip_color('some\x1b[Kthing') == 'something' # remove EL (erase to end of line)
assert strip_color('some\x1b[mthing') == 'something' # remove truncated style ending
assert strip_color('some\x1b[49;39mthing') == 'something' # remove odd color ends
def test_ansilen():
assert ansilen('this') == 4
assert ansilen('this\x1b[K\x1b[0m') == 4
assert ansilen(red('this')) == 4
assert ansilen(color('this', 12, 22, 'underline+blink')) == 4
def test_parse_rgb():
peachpuff = (255, 218, 185)
assert parse_rgb('peachpuff') == peachpuff
assert parse_rgb('#ffdab9') == peachpuff
assert parse_rgb('rgb(255, 218, 185)') == peachpuff
assert parse_rgb('peachpuff') == peachpuff
rebeccapurple = (102, 51, 153)
assert parse_rgb('rebeccapurple') == rebeccapurple
assert parse_rgb('#639') == rebeccapurple
assert parse_rgb('rgb(102,51,153)') == rebeccapurple
def test_partial_functions():
# a few spot-checks
assert black('test') == color('test', 'black')
assert red('test') == color('test', 'red')
assert green('test') == color('test', 'green')
assert bold('test') == color('test', style='bold')
assert underline('test') == color('test', style='underline')
def test_doc_example():
"""Text examples given in documentation"""
assert color('my string', fg='blue') == \
'\x1b[34mmy string\x1b[0m'
assert color('some text', fg='red', bg='yellow', style='underline') == \
'\x1b[31;43;4msome text\x1b[0m'
def test_custom_partial():
important = partial(color, fg='red', style='bold+underline')
boldul = partial(color, style='bold+underline')
text = 'very important'
answer = '\x1b[31;1;4mvery important\x1b[0m'
assert color(text, fg='red', style='bold+underline') == answer
assert red(text, style='bold+underline') == answer
assert boldul(text, fg='red') == answer
assert important(text) == answer
@pytest.mark.skipif(not _PY2, reason="only relevant for PY2")
def test_unicode_strings_py2():
"""
Ensure that colors, given a str, returns a str. We already
know from other tests that given a unicode, returns a unicode
(else they would fail gloriously).
"""
r1base = 'This is good' # note is str, not unicode
r1 = red(r1base)
assert r1 == '\x1b[31mThis is good\x1b[0m'
assert strip_color(r1) == r1base
assert isinstance(strip_color(r1), str)
assert ansilen(r1) == len(r1base)
assert isinstance(r1, str)
def test_unicode_strings():
r1base = u'This is good'
r1 = red(r1base)
assert r1 == u'\x1b[31mThis is good\x1b[0m'
assert strip_color(r1) == r1base
assert ansilen(r1) == len(r1base)
r2base = u'This\u2014is über good'
r2 = red(r2base)
assert r2 == u'\x1b[31mThis\u2014is \u00fcber good\x1b[0m'
assert strip_color(r2) == r2base
assert ansilen(r2) == len(r2base)
|
