mirror of https://github.com/Textualize/rich.git
99 lines
3.0 KiB
Python
99 lines
3.0 KiB
Python
from rich.segment import Segment
|
|
from rich.style import Style
|
|
|
|
|
|
def test_repr():
|
|
assert repr(Segment("foo")) == "Segment('foo', None)"
|
|
assert repr(Segment.control("foo")) == "Segment.control('foo', None)"
|
|
|
|
|
|
def test_line():
|
|
assert Segment.line() == Segment("\n")
|
|
|
|
|
|
def test_apply_style():
|
|
segments = [Segment("foo"), Segment("bar", Style(bold=True))]
|
|
assert Segment.apply_style(segments, None) is segments
|
|
assert list(Segment.apply_style(segments, Style(italic=True))) == [
|
|
Segment("foo", Style(italic=True)),
|
|
Segment("bar", Style(italic=True, bold=True)),
|
|
]
|
|
|
|
|
|
def test_split_lines():
|
|
lines = [Segment("Hello\nWorld")]
|
|
assert list(Segment.split_lines(lines)) == [[Segment("Hello")], [Segment("World")]]
|
|
|
|
|
|
def test_split_and_crop_lines():
|
|
assert list(
|
|
Segment.split_and_crop_lines([Segment("Hello\nWorld!\n"), Segment("foo")], 4)
|
|
) == [
|
|
[Segment("Hell"), Segment("\n", None)],
|
|
[Segment("Worl"), Segment("\n", None)],
|
|
[Segment("foo"), Segment(" ")],
|
|
]
|
|
|
|
|
|
def test_adjust_line_length():
|
|
line = [Segment("Hello", "foo")]
|
|
assert Segment.adjust_line_length(line, 10, style="bar") == [
|
|
Segment("Hello", "foo"),
|
|
Segment(" ", "bar"),
|
|
]
|
|
|
|
line = [Segment("H"), Segment("ello, World!")]
|
|
assert Segment.adjust_line_length(line, 5) == [Segment("H"), Segment("ello")]
|
|
|
|
line = [Segment("Hello")]
|
|
assert Segment.adjust_line_length(line, 5) == line
|
|
|
|
|
|
def test_get_line_length():
|
|
assert Segment.get_line_length([Segment("foo"), Segment("bar")]) == 6
|
|
|
|
|
|
def test_get_shape():
|
|
assert Segment.get_shape([[Segment("Hello")]]) == (5, 1)
|
|
assert Segment.get_shape([[Segment("Hello")], [Segment("World!")]]) == (6, 2)
|
|
|
|
|
|
def test_set_shape():
|
|
assert Segment.set_shape([[Segment("Hello")]], 10) == [
|
|
[Segment("Hello"), Segment(" ")]
|
|
]
|
|
assert Segment.set_shape([[Segment("Hello")]], 10, 2) == [
|
|
[Segment("Hello"), Segment(" ")],
|
|
[Segment(" " * 10)],
|
|
]
|
|
|
|
|
|
def test_simplify():
|
|
assert list(
|
|
Segment.simplify([Segment("Hello"), Segment(" "), Segment("World!")])
|
|
) == [Segment("Hello World!")]
|
|
assert list(
|
|
Segment.simplify(
|
|
[Segment("Hello", "red"), Segment(" ", "red"), Segment("World!", "blue")]
|
|
)
|
|
) == [Segment("Hello ", "red"), Segment("World!", "blue")]
|
|
assert list(Segment.simplify([])) == []
|
|
|
|
|
|
def test_filter_control():
|
|
segments = [Segment("foo"), Segment("bar", is_control=True)]
|
|
assert list(Segment.filter_control(segments)) == [Segment("foo")]
|
|
assert list(Segment.filter_control(segments, is_control=True)) == [
|
|
Segment("bar", is_control=True)
|
|
]
|
|
|
|
|
|
def test_strip_styles():
|
|
segments = [Segment("foo", Style(bold=True))]
|
|
assert list(Segment.strip_styles(segments)) == [Segment("foo", None)]
|
|
|
|
|
|
def test_strip_links():
|
|
segments = [Segment("foo", Style(bold=True, link="https://www.example.org"))]
|
|
assert list(Segment.strip_links(segments)) == [Segment("foo", Style(bold=True))]
|