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
|
"""The Test file for CLI helpers."""
import pytest
from sqlfluff.cli.helpers import LazySequence, pad_line, wrap_elem, wrap_field
@pytest.mark.parametrize(
"in_str,length,res",
[
("abc", 5, ["abc"]),
# Space wrap test
("how now brown cow", 10, ["how now", "brown cow"]),
# Harder wrap test
("A hippopotamus came for tea", 10, ["A hippopot", "amus came", "for tea"]),
# Harder wrap test, with a newline.
("A hippopotamus\ncame for tea", 10, ["A hippopot", "amus came", "for tea"]),
],
)
def test__cli__helpers__wrap_elem(in_str, length, res):
"""Test wrapping."""
str_list = wrap_elem(in_str, length)
assert str_list == res
def test__cli__helpers__wrap_field_a():
"""Test simple wrapping."""
dct = wrap_field("abc", "How Now Brown Cow", width=40)
assert dct["label_list"] == ["abc"]
assert dct["val_list"] == ["How Now Brown Cow"]
assert "sep_char" in dct
assert dct["lines"] == 1
assert dct["label_width"] == 3
def test__cli__helpers__wrap_field_b():
"""Test simple wrapping with overlap avoidance."""
dct = wrap_field("abc", "How Now Brown Cow", width=23)
assert dct["label_list"] == ["abc"]
assert dct["val_list"] == ["How Now Brown Cow"]
assert dct["label_width"] == 3
def test__cli__helpers__wrap_field_c():
"""Test simple wrapping."""
dct = wrap_field("how now brn cow", "How Now Brown Cow", width=25)
assert dct["label_list"] == ["how now", "brn cow"]
assert dct["label_width"] == 7
assert dct["val_list"] == ["How Now Brown", "Cow"]
assert dct["lines"] == 2
def test__cli__helpers__pad_line():
"""Test line padding."""
assert pad_line("abc", 5) == "abc "
assert pad_line("abcdef", 10, align="right") == " abcdef"
def test_cli__helpers__lazy_sequence():
"""Test the LazySequence."""
getter_run = False
def _get_sequence():
nonlocal getter_run
getter_run = True
return [1, 2, 3]
seq = LazySequence(_get_sequence)
# Check the sequence isn't called on instantiation.
assert not getter_run
# Fetch an item...
assert seq[2] == 3
# .. and that now it has run.
assert getter_run
# Check other methods work
assert len(seq) == 3
|
