fix: correctly visualize array type

ibis-project · Jan 21, 2023 · 26b0b3f · 26b0b3f
1 parent 506ce01
commit 26b0b3f
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 2 deletions.
diff --git a/ibis/expr/visualize.py b/ibis/expr/visualize.py
@@ -65,13 +65,13 @@ def get_label(node):
             label_fmt = '<<I>{}</I>: <B>{}</B>{}>'
         else:
             label_fmt = '<<I>{}</I>: <B>{}</B><BR ALIGN="LEFT" />:: {}>'
-        label = label_fmt.format(escape(nodename), escape(name), typename)
+        label = label_fmt.format(escape(nodename), escape(name), escape(typename))
     else:
         if isinstance(node, ops.TableNode):
             label_fmt = '<<B>{}</B>{}>'
         else:
             label_fmt = '<<B>{}</B><BR ALIGN="LEFT" />:: {}>'
-        label = label_fmt.format(escape(name), typename)
+        label = label_fmt.format(escape(name), escape(typename))
     return label
 
 

diff --git a/ibis/tests/expr/test_visualize.py b/ibis/tests/expr/test_visualize.py
@@ -162,3 +162,9 @@ def test_html_escape(with_graphviz):
     # representation. If an error is thrown, _repr_png_ returns None.
     expr = ibis.table([('<a & b>', ibis.expr.datatypes.Array('string'))])
     assert expr._repr_png_() is not None
+
+    expr = ibis.array([1, 2, 3])
+    assert expr._repr_png_() is not None
+
+    expr = ibis.array([1, 2, 3]).name("COL")
+    assert expr._repr_png_() is not None