Fix improper printing on ScaledLocations

src/bloqade/ir/control/field.py

@@ -109,7 +109,9 @@ def __str__(self):
         return f"ScaledLocations(value={str(self.value)})"
 
     def print_node(self):
-        return "ScaledLocations"
+        ## formatting location: scale pair:
+        tmp = {f"{key.value}": val for key, val in self.value.items()}
+        return f"ScaledLocations({str(tmp)})"
 
     def children(self):
         # can return list or dict

src/bloqade/ir/tree_print.py

@@ -63,10 +63,13 @@ def get_value(self):
 
 
 class Printer:
-    def __init__(self, p=MockPrinter()):
+    def __init__(self, p=None):
         self.charset = UnicodeCharSet() if unicode_enabled else ASCIICharSet()
         self.state = State()
-        self.p = p
+        if p is None:
+            self.p = MockPrinter()
+        else:
+            self.p = p
         self.max_tree_depth = max_tree_depth
 
     def should_print_annotation(self, children):

tests/test_field.py

@@ -64,7 +64,7 @@ def test_scal_loc():
     with pytest.raises(ValueError):
         ScaledLocations({(2, 3): 2})
 
-    assert x.print_node() == "ScaledLocations"
+    assert x.print_node() == "ScaledLocations({'1': 1.0, '2': 2.0})"
     assert x.children() == {"Location 1": cast(1.0), "Location 2": cast(2.0)}
 
     mystdout = StringIO()
@@ -74,7 +74,7 @@ def test_scal_loc():
 
     assert (
         mystdout.getvalue()
-        == "ScaledLocations\n"
+        == "ScaledLocations({'1': 1.0, '2': 2.0})\n"
         + "├─ Location 1\n"
         + "│  ⇒ Literal: 1.0\n"
         + "⋮\n"
@@ -102,7 +102,7 @@ def test_field():
     assert (
         mystdout.getvalue()
         == "Field\n"
-        + "└─ ScaledLocations\n"
+        + "└─ ScaledLocations({'1': 1.0, '2': 2.0})\n"
         + "   ⇒ Linear\n"
         + "     ├─ start\n"
         + "     │  ⇒ Literal: 1.0\n"
@@ -123,5 +123,7 @@ def test_field():
     assert f2.print_node() == "Field"
     # assert type(hash(f1)) == int
     assert f1.children() == {
-        "ScaledLocations": Linear(start=1.0, stop=cast("x"), duration=3.0)
+        "ScaledLocations({'1': 1.0, '2': 2.0})": Linear(
+            start=1.0, stop=cast("x"), duration=3.0
+        )
     }