add test

Signed-off-by: Future-Outlier <eric901201@gmail.com>
flyteorg · May 25, 2024 · 4733f0b · 4733f0b
1 parent c6eda1d
commit 4733f0b
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/tests/flytekit/unit/core/test_type_engine.py b/tests/flytekit/unit/core/test_type_engine.py
@@ -2275,8 +2275,12 @@ def t1(a: int) -> int:
         downstream_t(a=v, df=df)
 
         return v_1
+
+    @workflow
+    def wf(a: int) -> int:
+        return t1(a=a)
 
-    assert t1(a=3) == 9
+    assert wf(a=3) == 9
 
 
 def test_literal_hash_int_can_be_set():