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
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
|
import pytest
from terminaltexteffects.engine.motion import Coord
from terminaltexteffects.utils.graphics import Color, ColorPair, Gradient, random_color
pytestmark = [pytest.mark.utils, pytest.mark.smoke]
def test_random_color() -> None:
assert isinstance(random_color(), Color)
def test_color_pair_init() -> None:
cp = ColorPair("#ffffff", "#000000")
assert cp.fg_color == Color("#ffffff")
assert cp.bg_color == Color("#000000")
def test_color_pair_init_single_color() -> None:
cp = ColorPair("#ffffff")
assert cp.fg_color == Color("#ffffff")
assert cp.bg_color is None
def test_gradient_zero_stops() -> None:
with pytest.raises(ValueError):
Gradient()
def test_gradient_zero_steps() -> None:
with pytest.raises(ValueError):
Gradient(Color("#ffffff"), steps=0)
def test_gradient_zero_steps_tuple() -> None:
with pytest.raises(ValueError):
Gradient(Color("#ffffff"), Color("#000000"), Color("#ff0000"), steps=(1, 0))
def test_gradient_slice() -> None:
g = Gradient(Color("#ffffff"), Color("#000000"), steps=4)
assert g[0] == Color("#ffffff")
assert g[-1] == Color("#000000")
assert g[1:3] == [Color("#bfbfbf"), Color("#7f7f7f")]
def test_gradient_iter() -> None:
g = Gradient(Color("#ffffff"), Color("#000000"), steps=4)
for color in g:
assert isinstance(color, Color)
def test_gradient_str() -> None:
g = Gradient(Color("#ffffff"), Color("#000000"), steps=4)
assert "Stops(ffffff, 000000)" in str(g)
def test_gradient_len() -> None:
g = Gradient(Color("#ffffff"), Color("#000000"), steps=4)
assert len(g) == 5
def test_gradient_length_single_color() -> None:
g = Gradient(Color("#ffffff"), steps=5)
assert len(g.spectrum) == 5
def test_gradient_length_two_colors() -> None:
g = Gradient(Color("#000000"), Color("#ffffff"), steps=5)
assert len(g.spectrum) == 6
def test_gradient_length_three_colors() -> None:
g = Gradient(Color("#000000"), Color("#ffffff"), Color("#000000"), steps=5)
assert len(g.spectrum) == 11
def test_gradient_length_same_color_multiple_times() -> None:
g = Gradient(Color("#ffffff"), Color("#ffffff"), Color("#ffffff"), Color("#ffffff"), steps=4)
assert len(g.spectrum) == 13
def test_gradient_length_same_color_multiple_times_with_tuple_steps() -> None:
g = Gradient(Color("#ffffff"), Color("#ffffff"), Color("#ffffff"), Color("#ffffff"), steps=(4, 6))
assert len(g.spectrum) == 17
def test_gradient_single_color() -> None:
g = Gradient(Color("#ffffff"), steps=5)
assert all(color == Color("#ffffff") for color in g.spectrum)
def test_gradient_two_colors() -> None:
g = Gradient(Color("#000000"), Color("#ffffff"), steps=3)
assert g.spectrum[0] == Color("#000000") and g.spectrum[-1] == Color("#ffffff")
def test_gradient_single_step() -> None:
g = Gradient(Color("#ffffff"), steps=1)
assert g.spectrum[0] == Color("#ffffff")
def test_gradient_three_colors() -> None:
g = Gradient(Color("#ffffff"), Color("#000000"), Color("#ffffff"), steps=4)
assert (
g.spectrum[0] == Color("#ffffff") and g.spectrum[4] == Color("#000000") and g.spectrum[-1] == Color("#ffffff")
)
def test_gradient_loop() -> None:
g = Gradient(Color("#ffffff"), Color("#000000"), steps=4, loop=True)
assert g.spectrum[-1] == Color("#ffffff")
def test_gradient_get_color_at_fraction() -> None:
g = Gradient(Color("#ffffff"), Color("#000000"), steps=4)
assert g.get_color_at_fraction(0) == Color("#ffffff")
assert g.get_color_at_fraction(0.5) == Color("#7f7f7f")
assert g.get_color_at_fraction(1) == Color("#000000")
def test_gradient_get_color_at_fraction_invalid_float() -> None:
g = Gradient(Color("#ffffff"), Color("#000000"), steps=4)
with pytest.raises(ValueError):
g.get_color_at_fraction(1.1)
@pytest.mark.parametrize(
"direction",
[
Gradient.Direction.DIAGONAL,
Gradient.Direction.HORIZONTAL,
Gradient.Direction.VERTICAL,
Gradient.Direction.RADIAL,
],
)
def test_gradient_build_coordinate_color_mapping(direction) -> None:
g = Gradient(Color("#ffffff"), Color("#000000"), steps=4)
coordinate_map = g.build_coordinate_color_mapping(1, 10, 1, 10, direction)
if direction == Gradient.Direction.DIAGONAL:
assert coordinate_map[Coord(1, 1)] == Color("#ffffff")
assert coordinate_map[Coord(10, 10)] == Color("#000000")
elif direction == Gradient.Direction.HORIZONTAL:
assert coordinate_map[Coord(1, 1)] == Color("#ffffff")
assert coordinate_map[Coord(10, 1)] == Color("#000000")
elif direction == Gradient.Direction.VERTICAL:
assert coordinate_map[Coord(1, 1)] == Color("#ffffff")
assert coordinate_map[Coord(1, 10)] == Color("#000000")
elif direction == Gradient.Direction.RADIAL:
assert coordinate_map[Coord(5, 5)] == Color("#ffffff")
assert coordinate_map[Coord(10, 10)] == Color("#000000")
@pytest.mark.parametrize(
"direction",
[
Gradient.Direction.DIAGONAL,
Gradient.Direction.HORIZONTAL,
Gradient.Direction.VERTICAL,
Gradient.Direction.RADIAL,
],
)
@pytest.mark.parametrize("min_column", [1, 5])
@pytest.mark.parametrize("max_column", [5, 10])
@pytest.mark.parametrize("min_row", [1, 5])
@pytest.mark.parametrize("max_row", [5, 10])
def test_gradient_build_coordinate_color_mapping_no_exceptions(
direction,
min_column,
max_column,
min_row,
max_row,
) -> None:
g = Gradient(Color("#ffffff"), Color("#000000"), steps=4)
if min_column > max_column or min_row > max_row:
with pytest.raises(ValueError):
coordinate_map = g.build_coordinate_color_mapping(min_row, max_row, min_column, max_column, direction)
else: # check for exceptions across single row/column and issue that might arise from math calculations
coordinate_map = g.build_coordinate_color_mapping(min_row, max_row, min_column, max_column, direction)
def test_gradient_build_coordinate_color_mapping_single_row() -> None:
g = Gradient(Color("#ffffff"), Color("#000000"), steps=4)
coordinate_map = g.build_coordinate_color_mapping(1, 1, 1, 10, Gradient.Direction.HORIZONTAL)
assert coordinate_map[Coord(1, 1)] == Color("#ffffff")
assert coordinate_map[Coord(10, 1)] == Color("#000000")
def test_gradient_build_coordinate_color_mapping_single_column() -> None:
g = Gradient(Color("#ffffff"), Color("#000000"), steps=4)
coordinate_map = g.build_coordinate_color_mapping(1, 10, 1, 1, Gradient.Direction.VERTICAL)
assert coordinate_map[Coord(1, 1)] == Color("#ffffff")
assert coordinate_map[Coord(1, 10)] == Color("#000000")
def test_gradient_build_coordinate_color_mapping_single_row_column() -> None:
g = Gradient(Color("#ffffff"), Color("#000000"), steps=4)
coordinate_map = g.build_coordinate_color_mapping(1, 1, 1, 1, Gradient.Direction.HORIZONTAL)
assert coordinate_map[Coord(1, 1)] == Color("#000000")
def test_gradient_build_coordinate_color_mapping_invalid_row_column() -> None:
g = Gradient(Color("#ffffff"), Color("#000000"), steps=4)
with pytest.raises(ValueError):
g.build_coordinate_color_mapping(0, 10, 0, 10, Gradient.Direction.HORIZONTAL)
with pytest.raises(ValueError):
g.build_coordinate_color_mapping(10, 0, 10, 0, Gradient.Direction.HORIZONTAL)
def test_gradient_build_coordinate_color_mapping_max_less_than_min() -> None:
g = Gradient(Color("#ffffff"), Color("#000000"), steps=4)
with pytest.raises(ValueError):
g.build_coordinate_color_mapping(10, 1, 10, 1, Gradient.Direction.HORIZONTAL)
def test_color_invalid_xterm_color() -> None:
with pytest.raises(ValueError):
Color(256)
def test_color_invalid_hex_color() -> None:
with pytest.raises(ValueError):
Color("#ffffzz")
def test_color_valid_hex_with_hash():
assert Color("#ffffff") == Color("#ffffff")
def test_color_hex_rgb_ints():
assert Color("#000000").rgb_ints == (0, 0, 0)
def test_color_xterm_rgb_ints():
assert Color(0).rgb_ints == (0, 0, 0)
def test_color_not_equal():
assert Color("#ffffff") != Color("#000000")
def test_color_not_equal_different_types():
assert Color("#ffffff") != 0
assert Color(0) != "ffffff"
def test_color_is_hashable():
hash(Color("#ffffff"))
hash(Color(0))
def test_color_is_iterable():
assert list(Color("#ffffff")) == [Color("#ffffff")]
def test_color_repr():
assert repr(Color("#ffffff")) == "Color('ffffff')"
def test_color_str():
assert "Color Code: ffffff" in str(Color("#ffffff"))
|
