add typeengine check as well

Signed-off-by: Yee Hing Tong <[email protected]>
flyteorg · Oct 18, 2024 · 8f7035a · 8f7035a
1 parent a98c4c0
commit 8f7035a
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/tests/flytekit/unit/core/test_type_engine.py b/tests/flytekit/unit/core/test_type_engine.py
@@ -3702,3 +3702,6 @@ def test_structured_dataset_mismatch():
     transformer = TypeEngine.get_transformer(StructuredDataset)
     with pytest.raises(TypeTransformerFailedError):
         transformer.to_literal(FlyteContext.current_context(), df, StructuredDataset, TypeEngine.to_literal_type(StructuredDataset))
+
+    with pytest.raises(TypeTransformerFailedError):
+        TypeEngine.to_literal(FlyteContext.current_context(), df, StructuredDataset, TypeEngine.to_literal_type(StructuredDataset))