Add test case for elided binding if if/else node

Author: Lunderberg

tests/python/relax/test_tvmscript_parser.py

@@ -1865,6 +1865,36 @@ def func(x: R.Tensor((), "int32")) -> R.Tensor((), "int32"):
             return x
 
 
+def test_function_with_void_return_type_in_if_else():
+    """Last statement in if/else may be a void return"""
+
+    @I.ir_module
+    class Unsugared:
+        @R.function(pure=False)
+        def conditional(
+            x: R.Tensor((), "int32"), condition: R.Tensor((), "bool")
+        ) -> R.Tensor((), "int32"):
+            if condition:
+                y = R.print(x, format="True condition: {}")
+            else:
+                y = R.print(x, format="False condition: {}")
+            return x
+
+    @I.ir_module
+    class Sugared:
+        @R.function(pure=False)
+        def conditional(
+            x: R.Tensor((), "int32"), condition: R.Tensor((), "bool")
+        ) -> R.Tensor((), "int32"):
+            if condition:
+                R.print(x, format="True condition: {}")
+            else:
+                R.print(x, format="False condition: {}")
+            return x
+
+    tvm.ir.assert_structural_equal(Unsugared, Sugared)
+
+
 def test_call_pure_packed():
     @R.function
     def foo(x: R.Tensor((32, 32), "float32")) -> R.Tensor: