fix low codecov

Signed-off-by: Samhita Alla <[email protected]>
flyteorg · Jul 11, 2022 · 096b8ef · 096b8ef
1 parent edcefef
commit 096b8ef
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/tests/flytekit/unit/core/test_type_hints.py b/tests/flytekit/unit/core/test_type_hints.py
@@ -91,9 +91,10 @@ def my_task(a: int) -> typing.NamedTuple("OutputsBC", b=typing.ForwardRef("int")
 
 def test_annotated_namedtuple_output():
     @task
-    def my_task() -> typing.NamedTuple("OutputA", a=Annotated[int, "metadata-a"]):
-        return 1
+    def my_task(a: int) -> typing.NamedTuple("OutputA", a=Annotated[int, "metadata-a"]):
+        return a + 2
 
+    assert my_task(a=9) == (11,)
     assert get_origin(my_task.python_interface.outputs["a"]) is Annotated