Skip to content

Commit 9c0fbec

Browse files
Check that samples are packaged in build_spec tests
1 parent f8b1bc3 commit 9c0fbec

File tree

1 file changed

+7
-0
lines changed

1 file changed

+7
-0
lines changed

tests/build_spec/test_build_spec.py

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -106,6 +106,13 @@ def _test_build_spec(
106106
loaded_config = loaded_model.config
107107
assert "deepimagej" in loaded_config
108108

109+
if loaded_model.sample_inputs is not missing:
110+
for sample in loaded_model.sample_inputs:
111+
assert sample.exists()
112+
if loaded_model.sample_outputs is not missing:
113+
for sample in loaded_model.sample_outputs:
114+
assert sample.exists()
115+
109116
attachments = loaded_model.attachments
110117
if attachments is not missing and attachments.files is not missing:
111118
for attached_file in attachments["files"]:

0 commit comments

Comments
 (0)