Fixed test

Signed-off-by: Kevin Su <[email protected]>
flyteorg · Oct 25, 2021 · 32e5eed · 32e5eed
1 parent 994437a
commit 32e5eed
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/tests/flytekit/unit/core/test_type_hints.py b/tests/flytekit/unit/core/test_type_hints.py
@@ -1425,5 +1425,9 @@ def foo3(a: typing.Dict) -> typing.Dict:
     with pytest.raises(TypeError, match="Failed to convert return value for var o0 for function test_type_hints.foo2"):
         foo2(a=10, b="hello")
 
-    with pytest.raises(TypeError, match="Not a collection type simple: STRUCT\n but got a list \\[{'hello': 2}\\]"):
+    with pytest.raises(
+        TypeError,
+        match="Not a collection type simple: STRUCT\nmetadata {\n  fields {\n    key: \"python_class_name\"\n"
+              "    value {\n      string_value: \"VT\"\n    }\n  }\n}\n but got a list [{\'hello\': 2}]",
+    ):
         foo3(a=[{"hello": 2}])