add tests for arrange

2025-10-17 02:38:12 +03:00 · 2022-07-29 17:10:43 +01:00
parent 9aecc86152
commit 52199ab134
3 changed files with 118 additions and 9 deletions
--- a/tests/test_arrange.py
+++ b/tests/test_arrange.py
@@ -0,0 +1,109 @@
+from textual._arrange import arrange, TOP_Z
+from textual._layout import WidgetPlacement
+from textual.geometry import Region, Size, Spacing
+from textual.widget import Widget
+
+
+def test_arrange_empty():
+    container = Widget(id="container")
+
+    placements, widgets, spacing = arrange(container, [], Size(80, 24), Size(80, 24))
+    assert placements == []
+    assert widgets == set()
+    assert spacing == Spacing(0, 0, 0, 0)
+
+
+def test_arrange_dock_top():
+    container = Widget(id="container")
+    child = Widget(id="child")
+    header = Widget(id="header")
+    header.styles.dock = "top"
+    header.styles.height = "1"
+
+    placements, widgets, spacing = arrange(
+        container, [child, header], Size(80, 24), Size(80, 24)
+    )
+    assert placements == [
+        WidgetPlacement(Region(0, 0, 80, 1), header, order=TOP_Z, fixed=True),
+        WidgetPlacement(Region(0, 1, 80, 23), child, order=0, fixed=False),
+        WidgetPlacement(
+            region=Region(x=0, y=1, width=80, height=23),
+            widget=None,
+            order=0,
+            fixed=False,
+        ),
+    ]
+    assert widgets == {child, header}
+    assert spacing == Spacing(1, 0, 0, 0)
+
+
+def test_arrange_dock_left():
+    container = Widget(id="container")
+    child = Widget(id="child")
+    header = Widget(id="header")
+    header.styles.dock = "left"
+    header.styles.width = "10"
+
+    placements, widgets, spacing = arrange(
+        container, [child, header], Size(80, 24), Size(80, 24)
+    )
+    assert placements == [
+        WidgetPlacement(Region(0, 0, 10, 24), header, order=TOP_Z, fixed=True),
+        WidgetPlacement(Region(10, 0, 70, 24), child, order=0, fixed=False),
+        WidgetPlacement(
+            region=Region(x=10, y=0, width=70, height=24),
+            widget=None,
+            order=0,
+            fixed=False,
+        ),
+    ]
+    assert widgets == {child, header}
+    assert spacing == Spacing(0, 0, 0, 10)
+
+
+def test_arrange_dock_right():
+    container = Widget(id="container")
+    child = Widget(id="child")
+    header = Widget(id="header")
+    header.styles.dock = "right"
+    header.styles.width = "10"
+
+    placements, widgets, spacing = arrange(
+        container, [child, header], Size(80, 24), Size(80, 24)
+    )
+    assert placements == [
+        WidgetPlacement(Region(70, 0, 10, 24), header, order=TOP_Z, fixed=True),
+        WidgetPlacement(Region(0, 0, 70, 24), child, order=0, fixed=False),
+        WidgetPlacement(
+            region=Region(x=0, y=0, width=70, height=24),
+            widget=None,
+            order=0,
+            fixed=False,
+        ),
+    ]
+    assert widgets == {child, header}
+    assert spacing == Spacing(0, 10, 0, 0)
+
+
+def test_arrange_dock_bottom():
+    container = Widget(id="container")
+    child = Widget(id="child")
+    header = Widget(id="header")
+    header.styles.dock = "bottom"
+    header.styles.height = "1"
+
+    placements, widgets, spacing = arrange(
+        container, [child, header], Size(80, 24), Size(80, 24)
+    )
+    assert placements == [
+        WidgetPlacement(Region(0, 23, 80, 1), header, order=TOP_Z, fixed=True),
+        WidgetPlacement(Region(0, 0, 80, 23), child, order=0, fixed=False),
+        WidgetPlacement(
+            region=Region(x=0, y=0, width=80, height=23),
+            widget=None,
+            order=0,
+            fixed=False,
+        ),
+    ]
+    assert widgets == {child, header}
+    assert spacing == Spacing(0, 0, 1, 0)