1 files changed, 117 insertions, 0 deletions
diff --git a/tests/test_segment.py b/tests/test_segment.py
new file mode 100644
index 0000000..7e7a3b3
--- /dev/null
+++ b/tests/test_segment.py
@@ -0,0 +1,117 @@
+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))]
+
+
+def test_remove_color():
+    segments = [
+        Segment("foo", Style(bold=True, color="red")),
+        Segment("bar", None),
+    ]
+    assert list(Segment.remove_color(segments)) == [
+        Segment("foo", Style(bold=True)),
+        Segment("bar", None),
+    ]
+
+
+def test_make_control():
+    segments = [Segment("foo"), Segment("bar")]
+    assert Segment.make_control(segments) == [
+        Segment.control("foo"),
+        Segment.control("bar"),
+    ]