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
|
import pytest
from textual._wrap import chunks, compute_wrap_offsets
@pytest.mark.parametrize(
"input_text, expected_output",
[
("", []),
(" ", [(0, 4, " ")]),
("\t", [(0, 1, "\t")]),
("foo", [(0, 3, "foo")]),
(" foo ", [(0, 2, " "), (2, 7, "foo ")]),
("foo bar", [(0, 4, "foo "), (4, 7, "bar")]),
("\tfoo bar", [(0, 1, "\t"), (1, 5, "foo "), (5, 8, "bar")]),
(" foo bar", [(0, 1, " "), (1, 5, "foo "), (5, 8, "bar")]),
("foo bar ", [(0, 4, "foo "), (4, 10, "bar ")]),
("foo\t bar ", [(0, 6, "foo\t "), (6, 12, "bar ")]),
("木\t 川 ", [(0, 4, "木\t "), (4, 8, "川 ")]),
],
)
def test_chunks(input_text, expected_output):
assert list(chunks(input_text)) == expected_output
@pytest.mark.parametrize(
"text, width, tab_size, expected_output",
[
("", 6, 4, []),
("\t", 6, 4, []),
(" ", 6, 4, []),
("foo bar baz", 6, 4, [4, 8]),
("\tfoo bar baz", 6, 4, [1, 5, 9]),
("\tfo bar baz", 6, 4, [1, 4, 8]),
("\tfo bar baz", 6, 8, [1, 4, 8]),
("\tfo bar baz\t", 6, 8, [1, 4, 8]),
("\t\t\tfo bar baz\t", 20, 4, [10]),
("\t\t\t\t\t\t\t\tfo bar bar", 19, 4, [4, 11]),
("\t\t\t\t\t", 19, 4, [4]),
],
)
def test_compute_wrap_offsets(text, width, tab_size, expected_output):
assert compute_wrap_offsets(text, width, tab_size) == expected_output
|
