Skip to content

Commit fc3fddd

Browse files
committed
Add more thorough asserts for dataset create/delete test, and add a dataset fixture
1 parent 543f02b commit fc3fddd

File tree

1 file changed

+14
-0
lines changed

1 file changed

+14
-0
lines changed

tests/test_dataset.py

Lines changed: 14 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -6,10 +6,24 @@
66

77
TEST_DATASET_NAME = '[PyTest] Test Dataset'
88

9+
@pytest.fixture(scope='module')
10+
def dataset(CLIENT):
11+
ds = CLIENT.create_dataset(TEST_DATASET_NAME)
12+
yield ds
13+
14+
CLIENT.delete_dataset(ds.id)
15+
16+
917
def test_dataset_create_and_delete(CLIENT):
1018
# Creation
1119
ds = CLIENT.create_dataset(TEST_DATASET_NAME)
1220
assert isinstance(ds, Dataset)
21+
assert ds.name == TEST_DATASET_NAME
22+
assert ds.model_runs == []
23+
assert ds.slices == []
24+
assert ds.size == 0
25+
assert ds.items == []
1326

1427
# Deletion
1528
response = CLIENT.delete_dataset(ds.id)
29+
assert response == {}

0 commit comments

Comments
 (0)