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
|
from collections import UserList, deque
from typing import Sequence
import pytest
from tests.utilities.render import render
from textual.renderables.sparkline import Sparkline
GREEN = "\x1b[38;2;0;255;0m"
RED = "\x1b[38;2;255;0;0m"
BLENDED = "\x1b[38;2;127;127;0m" # Color between red and green
STOP = "\x1b[0m"
def test_sparkline_no_data():
assert render(Sparkline([], width=4)) == f"{GREEN}▁▁▁▁{STOP}"
def test_sparkline_single_datapoint():
assert render(Sparkline([2.5], width=4)) == f"{RED}████{STOP}"
def test_sparkline_two_values_min_max():
print(repr(render(Sparkline([2, 4], width=2))))
assert render(Sparkline([2, 4], width=2)) == f"{GREEN}▁{STOP}{RED}█{STOP}"
def test_sparkline_expand_data_to_width():
assert (
render(Sparkline([2, 4], width=4))
== f"{GREEN}▁{STOP}{GREEN}▁{STOP}{RED}█{STOP}{RED}█{STOP}"
)
def test_sparkline_expand_data_to_width_non_divisible():
assert (
render(Sparkline([2, 4], width=3))
== f"{GREEN}▁{STOP}{GREEN}▁{STOP}{RED}█{STOP}"
)
def test_sparkline_shrink_data_to_width():
assert (
render(Sparkline([2, 2, 4, 4, 6, 6], width=3))
== f"{GREEN}▁{STOP}{BLENDED}▄{STOP}{RED}█{STOP}"
)
def test_sparkline_color_blend():
assert (
render(Sparkline([1, 2, 3], width=3))
== f"{GREEN}▁{STOP}{BLENDED}▄{STOP}{RED}█{STOP}"
)
@pytest.mark.parametrize(
"data",
[
(1, 2, 3),
[1, 2, 3],
bytearray((1, 2, 3)),
bytes((1, 2, 3)),
deque([1, 2, 3]),
range(1, 4),
UserList((1, 2, 3)),
],
)
def test_sparkline_sequence_types(data: Sequence[int]):
"""Sparkline should work with common Sequence types."""
assert issubclass(type(data), Sequence)
assert (
render(Sparkline(data, width=3))
== f"{GREEN}▁{STOP}{BLENDED}▄{STOP}{RED}█{STOP}"
)
|
