Skip to content

Commit 1760e07

Browse files
committed
chore: tests
Signed-off-by: ThibaultFy <[email protected]>
1 parent e60a813 commit 1760e07

File tree

1 file changed

+20
-0
lines changed

1 file changed

+20
-0
lines changed

tests/sdk/test_schemas.py

Lines changed: 20 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -4,6 +4,8 @@
44
import pytest
55

66
from substra.sdk.schemas import DataSampleSpec
7+
from substra.sdk.schemas import DatasetSpec
8+
from substra.sdk.schemas import Permissions
79

810

911
@pytest.mark.parametrize("path", [pathlib.Path() / "data", "./data", pathlib.Path().cwd() / "data"])
@@ -38,3 +40,21 @@ def test_datasample_spec_paths_set_to_none():
3840
def test_datasample_spec_path_set_to_none():
3941
with pytest.raises(ValueError):
4042
DataSampleSpec(path=None, data_manager_keys=[str(uuid.uuid4())])
43+
44+
45+
def test_dataset_spec_no_description(tmpdir):
46+
47+
opener_path = tmpdir / "opener.py"
48+
with open(opener_path, "w") as f:
49+
f.write("print('I'm opening your data')")
50+
51+
permissions = Permissions(public=True, authorized_ids=[])
52+
53+
DatasetSpec(
54+
name="Fake Dataset",
55+
data_opener=str(opener_path),
56+
permissions=permissions,
57+
logs_permission=permissions,
58+
)
59+
60+
assert (pathlib.Path(opener_path).parent / "generated_description.md").exists

0 commit comments

Comments
 (0)