Add tests to improve codecov coverage for FlyteFile and FlyteDirectory validators

Signed-off-by: Govert Verkes <govert@cusp.ai>

Author: gverkes

tests/flytekit/unit/extras/pydantic_transformer/test_pydantic_basemodel_transformer.py

@@ -1041,6 +1041,61 @@ class BM(BaseModel):
     bm2.model_dump()
 
 
+def test_flytefile_pydantic_with_local_file(local_dummy_file):
+    class BM(BaseModel):
+        ff: FlyteFile
+
+    bm = BM(ff=FlyteFile(path=local_dummy_file))
+
+    bm_dict = bm.model_dump()
+    bm2 = BM.model_validate(bm_dict)
+
+    assert isinstance(bm2.ff, FlyteFile)
+    assert hasattr(bm2.ff, "_downloader")
+    assert hasattr(bm2.ff, "_remote_source")
+
+    bm2.model_dump()
+
+
+def test_flytefile_pydantic_with_metadata(local_dummy_file):
+    class BM(BaseModel):
+        ff: FlyteFile
+
+    bm = BM(ff=FlyteFile(path=local_dummy_file, metadata={"key": "value"}))
+
+    bm_dict = bm.model_dump()
+    bm2 = BM.model_validate(bm_dict)
+
+    assert isinstance(bm2.ff, FlyteFile)
+    assert hasattr(bm2.ff, "_downloader")
+    assert hasattr(bm2.ff, "_remote_source")
+    assert bm2.ff.metadata == {"key": "value"}
+
+    bm2.model_dump()
+
+
+def test_flytefile_pydantic_direct_dict_validate(local_dummy_file):
+    class BM(BaseModel):
+        ff: FlyteFile
+
+    bm = BM.model_validate({"ff": {"path": local_dummy_file}})
+
+    assert isinstance(bm.ff, FlyteFile)
+    assert hasattr(bm.ff, "_downloader")
+    assert hasattr(bm.ff, "_remote_source")
+
+
+def test_flytedirectory_pydantic_direct_dict_validate(local_dummy_directory):
+    class BM(BaseModel):
+        fd: FlyteDirectory
+
+    bm = BM.model_validate({"fd": {"path": local_dummy_directory}})
+
+    assert isinstance(bm.fd, FlyteDirectory)
+    assert hasattr(bm.fd, "_downloader")
+    assert hasattr(bm.fd, "_remote_source")
+
+
 def test_flytedirectory_pydantic_model_dump_validate_cycle():
     class BM(BaseModel):
         fd: FlyteDirectory
@@ -1056,3 +1111,19 @@ class BM(BaseModel):
     assert bm2.fd.remote_source == "s3://my-bucket/my-dir"
 
     bm2.model_dump()
+
+
+def test_flytedirectory_pydantic_with_local_directory(local_dummy_directory):
+    class BM(BaseModel):
+        fd: FlyteDirectory
+
+    bm = BM(fd=FlyteDirectory(path=local_dummy_directory))
+
+    bm_dict = bm.model_dump()
+    bm2 = BM.model_validate(bm_dict)
+
+    assert isinstance(bm2.fd, FlyteDirectory)
+    assert hasattr(bm2.fd, "_downloader")
+    assert hasattr(bm2.fd, "_remote_source")
+
+    bm2.model_dump()