test for serialization

Author: leila79

python/rlc/layout_logger.py

@@ -156,13 +156,13 @@ def rec(node):
         rec(root)
         self._node_meta.clear()
 
-    def to_json(self, indent: int = 2) -> str:
+    def to_json(self) -> str:
         data = {
             "config" : asdict(self.config),
             "events" : self._events,
             "final_tree" : self._final_tree
         }
-        return json.dumps(data, indent)
+        return json.dumps(data)
 
     def to_text_tree(self, root):
         lines : List[str] = []

python/test/test_serialization.py

@@ -0,0 +1,60 @@
+import json
+from pathlib import Path
+import pygame
+import pytest
+from rlc import Layout, Text, Padding, Direction, FIXED, FIT
+from rlc import LayoutLogger, LayoutLogConfig
+
+@pytest.fixture(autouse=True, scope="session")
+def init_pygame():
+    pygame.init()
+    pygame.font.init()
+    yield
+    pygame.quit()
+
+def build_simple_tree():
+    root = Layout("white", sizing=(FIT(), FIT()), padding=Padding(5, 5, 5, 5), direction=Direction.ROW)
+    child1 = Layout("blue", sizing=(FIXED(100), FIXED(50)))
+    child2 = Text("Hello World", "Arial", 16)
+    root.add_child(child1)
+    root.add_child(child2)
+    return root
+
+
+@pytest.mark.parametrize("indent", [2, 4])
+def test_json_and_text_serialization(indent, tmp_path):
+    root = build_simple_tree()
+    logger = LayoutLogger(LayoutLogConfig())
+
+    # compute and layout to fill logger events
+    root.compute_size(logger=logger)
+    root.layout(0, 0, logger=logger)
+    logger.record_final_tree(root)
+
+    # --- JSON string ---
+    js = logger.to_json()
+    data = json.loads(js)
+    assert "config" in data
+    assert "events" in data
+    assert "final_tree" in data
+    assert data["final_tree"]["type"] == "Layout"
+
+    # --- Write JSON file ---
+    json_path = tmp_path / "layout.json"
+    logger.write_json(str(json_path), indent=indent)
+    assert json_path.exists()
+    parsed = json.loads(json_path.read_text())
+    assert parsed["final_tree"]["type"] == "Layout"
+
+    # --- Text tree string ---
+    txt = logger.to_text_tree(root)
+    assert "Layout" in txt
+    assert "Text" in txt
+
+    # --- Write text tree file ---
+    txt_path = tmp_path / "layout.txt"
+    logger.write_text_tree(root, str(txt_path))
+    assert txt_path.exists()
+    contents = txt_path.read_text()
+    assert "Layout" in contents
+    assert "Text" in contents