Skip to content

Commit 7e4fd6f

Browse files
committed
Updated unit test.
1 parent beabbe3 commit 7e4fd6f

File tree

1 file changed

+3
-4
lines changed

1 file changed

+3
-4
lines changed

tests/unitary/default_setup/model/test_artifact_downloader.py

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -100,20 +100,19 @@ def test_downaload_small_artifact(self):
100100
expected_artifact_bytes_content
101101
)
102102
with tempfile.TemporaryDirectory() as tmp_dir:
103-
target_dir = os.path.join(tmp_dir, "model_artifacts/")
104103
SmallArtifactDownloader(
105104
dsc_model=self.mock_dsc_model,
106-
target_dir=target_dir,
105+
target_dir=tmp_dir,
107106
force_overwrite=True,
108107
).download()
109108

110109
self.mock_dsc_model.get_model_artifact_content.assert_called()
111110

112111
test_files = list(
113-
glob.iglob(os.path.join(target_dir, "**"), recursive=True)
112+
glob.iglob(os.path.join(tmp_dir, "**"), recursive=True)
114113
)
115114
expected_files = [
116-
os.path.join(target_dir, file_name)
115+
os.path.join(tmp_dir, file_name)
117116
for file_name in ["", "runtime.yaml", "score.py"]
118117
]
119118
assert sorted(test_files) == sorted(expected_files)

0 commit comments

Comments
 (0)