feat(tree): add vis method for printing tree in address order and update ParseResult to utilise it

Author: incrementaliser

src/dylan/tree/tree.py

@@ -75,6 +75,12 @@ def clone(self) -> Tree:
         """Deep copy nodes, pointer, and variable pools (Java ``Tree.clone`` sketch)."""
         return Tree(self)
 
+    def vis(self) -> None:
+        """Print this tree in address order (same plain-text view as :meth:`~dynamicsyntax.parse_result.ParseResult.vis`)."""
+        from dylan.gui.formatting import format_ds_tree
+
+        print(format_ds_tree(self))
+
     def set_pointer(self, addr: NodeAddress) -> None:
         """Set the pointer address (Java ``setPointer``)."""
         self.pointer = addr

src/dynamicsyntax/parse_result.py

@@ -48,7 +48,7 @@ def vis(self) -> None:
         if self.tree is None:
             print("(no parse tree)")
             return
-        print(format_ds_tree(self.tree))
+        self.tree.vis()
 
     def get_vocab(
         self,

tests/test_dynamicsyntax_parse.py

@@ -176,6 +176,17 @@ def test_vis_prints_address_order(capsys: pytest.CaptureFixture[str]) -> None:
     assert "00" in out and "man" in out.lower()
 
 
+def test_tree_vis_matches_parse_result_vis(capsys: pytest.CaptureFixture[str]) -> None:
+    """``Tree.vis`` prints the same address-order text as ``ParseResult.vis`` (e.g. after ``complete_tree``)."""
+    p = ds.parse("a man arrives", "ttr")
+    assert p.tree is not None
+    p.vis()
+    from_parse_result = capsys.readouterr().out
+    p.tree.vis()
+    from_tree = capsys.readouterr().out
+    assert from_parse_result == from_tree
+
+
 def test_parse_trace_snapshots() -> None:
     """``trace=True`` records one tree per word plus the initial state."""
     p = ds.parse("a man arrives", "ttr", trace=True)