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
|
"""Test the beams effect with various configuration arguments."""
from __future__ import annotations
from typing import TYPE_CHECKING
import pytest
from terminaltexteffects.effects import effect_beams
if TYPE_CHECKING:
from terminaltexteffects import Color, Gradient
from terminaltexteffects.engine.terminal import TerminalConfig
@pytest.mark.parametrize(
"input_data",
["empty", "single_char", "single_column", "single_row", "medium", "tabs"],
indirect=True,
)
def test_beams_effect(input_data: str, terminal_config_default_no_framerate: TerminalConfig) -> None:
"""Test the beams effect with various input data and default terminal configuration."""
effect = effect_beams.Beams(input_data)
effect.terminal_config = terminal_config_default_no_framerate
with effect.terminal_output() as terminal:
for frame in effect:
terminal.print(frame)
@pytest.mark.parametrize("input_data", ["medium"], indirect=True)
def test_beams_effect_terminal_color_options(
input_data: str,
terminal_config_with_color_options: TerminalConfig,
) -> None:
"""Test the beams effect with terminal color options."""
effect = effect_beams.Beams(input_data)
effect.terminal_config = terminal_config_with_color_options
with effect.terminal_output() as terminal:
for frame in effect:
terminal.print(frame)
@pytest.mark.parametrize("input_data", ["medium"], indirect=True)
def test_beams_final_gradient(
terminal_config_default_no_framerate: TerminalConfig,
input_data: str,
gradient_direction: Gradient.Direction,
gradient_steps: tuple[int, ...],
gradient_stops: tuple[Color, ...],
) -> None:
"""Test the final gradient configuration of the beams effect."""
effect = effect_beams.Beams(input_data)
effect.terminal_config = terminal_config_default_no_framerate
effect.effect_config.final_gradient_direction = gradient_direction
effect.effect_config.final_gradient_steps = gradient_steps
effect.effect_config.final_gradient_stops = gradient_stops
with effect.terminal_output() as terminal:
for frame in effect:
terminal.print(frame)
@pytest.mark.parametrize("beam_row_symbols", [("a",), ("a", "b", "c")])
@pytest.mark.parametrize("beam_column_symbols", [("a",), ("a", "b", "c")])
@pytest.mark.parametrize("beam_delay", [1, 3])
@pytest.mark.parametrize("beam_row_speed_range", [(1, 3), (2, 4)])
@pytest.mark.parametrize("beam_column_speed_range", [(1, 3), (2, 4)])
@pytest.mark.parametrize("input_data", ["single_char", "medium"], indirect=True)
def test_beams_effect_args(
input_data: str,
terminal_config_default_no_framerate: TerminalConfig,
beam_row_symbols: tuple[str, ...],
beam_column_symbols: tuple[str, ...],
beam_delay: int,
beam_row_speed_range: tuple[int, int],
beam_column_speed_range: tuple[int, int],
gradient_stops: tuple[Color, ...],
gradient_steps: tuple[int, ...],
gradient_frames: int,
) -> None:
"""Test the beams effect with various configuration arguments."""
effect = effect_beams.Beams(input_data)
effect.terminal_config = terminal_config_default_no_framerate
effect.effect_config.beam_row_symbols = beam_row_symbols
effect.effect_config.beam_column_symbols = beam_column_symbols
effect.effect_config.beam_delay = beam_delay
effect.effect_config.beam_row_speed_range = beam_row_speed_range
effect.effect_config.beam_column_speed_range = beam_column_speed_range
effect.effect_config.beam_gradient_stops = gradient_stops
effect.effect_config.beam_gradient_steps = gradient_steps
effect.effect_config.beam_gradient_frames = gradient_frames
with effect.terminal_output() as terminal:
for frame in effect:
terminal.print(frame)
|
