Add test case

Change-Id: I579b20da3f5d49054823392be80aaf78a055f596
apache · May 29, 2020 · 53b4198 · 53b4198
1 parent fb5f7ff
commit 53b4198
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/tests/python/relay/test_ir_text_printer.py b/tests/python/relay/test_ir_text_printer.py
@@ -240,6 +240,15 @@ def @main[A]() -> fn (A, List[A]) -> List[A] {
     assert main_def_str.strip() in mod_str
 
 
+def test_null_attribute():
+    x = relay.var("x")
+    y = relay.var("y")
+    z = relay.Function([x], y)
+    z = z.with_attr("TestAttribute", None)
+    txt = astext(z)
+    assert "TestAttribute=(nullptr)" in txt
+
+
 if __name__ == "__main__":
     do_print[0] = True
     test_lstm()
@@ -262,3 +271,4 @@ def @main[A]() -> fn (A, List[A]) -> List[A] {
     test_variable_name()
     test_call_node_order()
     test_unapplied_constructor()
+    test_null_attribute()