Fixed tests

Signed-off-by: Kevin Su <pingsutw@apache.org>
flyteorg · Dec 6, 2021 · 7288a2c · 7288a2c
1 parent a614ba6
commit 7288a2c
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/flytekit/unit/core/test_type_hints.py b/tests/flytekit/unit/core/test_type_hints.py
@@ -378,7 +378,7 @@ def wf(path: str) -> os.PathLike:
         n1 = t1(path=path)
         return t2(fs=n1)
 
-    assert wf(path="s3://somewhere") == "s3://somewhere"
+    assert "/tmp/flyte/" in wf(path="s3://somewhere").path
 
 
 def test_flyte_directory_in_dataclass():
@@ -409,7 +409,7 @@ def wf(path: str) -> os.PathLike:
         n1 = t1(path=path)
         return t2(fs=n1)
 
-    assert wf(path="s3://somewhere") == "s3://somewhere"
+    assert "/tmp/flyte/" in wf(path="s3://somewhere").path
 
 
 def test_wf1_with_map():