Added to_lol, eq and hash to Tree

Author: mapio

src/liblet/display.py

@@ -293,12 +293,24 @@ def _to_tree(ast_node):
 
     return _to_tree(node)
 
+  def to_lol(self):
+    def walk(T):
+      return (T.root, *tuple(walk(child) for child in T.children))
+
+    return walk(self)
+
   def __repr__(self):
     def walk(T):
       return '({}: {})'.format(T.root, ', '.join(map(walk, T.children))) if T.children else f'({T.root})'
 
     return walk(self)
 
+  def __eq__(self, other):
+    return isinstance(other, Tree) and self.to_lol() == other.to_lol()
+
+  def __hash__(self):
+    return hash(self.to_lol())
+
   def _gv_graph_(self):
     G = GVWrapper(
       dict(  # noqa: C408
@@ -712,7 +724,8 @@ def _fmt(r, c):
           letstr(self.data[r], self.fmt['elem_sep'], sort=self.fmt['letstr_sort'], remove_outer=True),
         )
         for r in rows
-      ), True
+      ),
+      True,
     )
 
 
@@ -735,7 +748,8 @@ def _repr_html_(self):
         )
         + '</pre></td>'
         for l in range(N, L - 1, -1)  # noqa: E741
-      ), True
+      ),
+      True,
     )
 
 
@@ -777,8 +791,13 @@ def embed_css(custom_css=CUSTOM_CSS):
   return HTML(f'<style>{custom_css}</style>')
 
 
-def liblet_table(content, as_str = False):
-  return f'<table class="liblet" data-quarto-disable-processing="true">{content}</table>' if as_str else HTML(f'<table class=liblet>{content}</table>')
+def liblet_table(content, as_str=False):
+  return (
+    f'<table class="liblet" data-quarto-disable-processing="true">{content}</table>'
+    if as_str
+    else HTML(f'<table class=liblet>{content}</table>')
+  )
+
 
 def resized_svg_repr(obj, width=800, height=600):
   if hasattr(obj, '_repr_image_svg_xml'):

src/tests/display_test.py

@@ -0,0 +1,21 @@
+import unittest
+
+from liblet.display import Tree
+
+
+class TestTree(unittest.TestCase):
+  def test_from_lol(self):
+    lol = (1, (11,), (12, (121,), (122,)))
+    t = Tree.from_lol(lol)
+    self.assertEqual(str(t), '(1: (11), (12: (121), (122)))')
+
+  def test_to_lol(self):
+    lol = (1, (11,), (12, (121,), (122,)))
+    t = Tree.from_lol(lol)
+    self.assertEqual(t.to_lol(), lol)
+
+  def test_equals(self):
+    lol = (1, (11,), (12, (121,), (122,)))
+    t1 = Tree.from_lol(lol)
+    t2 = Tree.from_lol(lol)
+    self.assertEqual(t1, t2)