fix tests

Signed-off-by: Kevin Su <[email protected]>

Author: pingsutw

tests/flytekit/unit/core/test_generice_idl_type_engine.py

@@ -2704,7 +2704,7 @@ def test_union_in_dataclass():
     lv = tf.to_literal(ctx, o, pt, lt)
     ot = tf.to_python_value(ctx, lv=lv, expected_python_type=pt)
 
-    return o == ot
+    assert o == ot
     assert o.result.schema.remote_path == ot.result.schema.remote_path
     assert o.result.number == ot.result.number
     assert o.schema.remote_path == ot.schema.remote_path

tests/flytekit/unit/core/test_type_engine.py

@@ -2713,7 +2713,7 @@ def test_union_in_dataclass():
     lv = tf.to_literal(ctx, o, pt, lt)
     ot = tf.to_python_value(ctx, lv=lv, expected_python_type=pt)
 
-    return o == ot
+    assert o == ot
     assert o.result.schema.remote_path == ot.result.schema.remote_path
     assert o.result.number == ot.result.number
     assert o.schema.remote_path == ot.schema.remote_path