def test_crud_datasets_metadata(client: PlatformClient):
dataset = 'blueno::test_crud_datasets_metadata'
assert dataset not in {d['name'] for d in client.list_datasets()}
client.create_dataset(dataset, description='This data is bad')
matches = [d for d in client.list_datasets() if d['name'] == dataset]
assert len(matches) == 1
assert matches[0]['name'] == dataset
assert matches[0]['info']['description'] == 'This data is bad'
client.delete_dataset(dataset)
matches = [d for d in client.list_datasets() if d['name'] == dataset]
assert len(matches) == 0
def test_crud_samples(client: PlatformClient):
dataset = 'blueno::test_crud_samples'
samples = [
'smaple1',
'snapple2',
'water3',
]
client.create_dataset(dataset)
# Attempt to create sample w/o data should pass
assert client.register_sample(
samples[0],
dataset,
data_url='file://test/crud_samples/no-data.xzx',
validate=False,
split='training')
assert len(client.list_samples(dataset)) == 1
# 2nd attempt to create w/ sample name should fail
assert not client.register_sample(
samples[0],
dataset,
data_url='file://test/crud_samples/no-data.xzx',
validate=False,
split='test')
listed_samples = client.list_samples(dataset)
assert len(listed_samples) == 1
# 2nd attempt to create w/ sample name should not change 'info'
assert listed_samples[0]['info']['split'] == 'training'
# Attempt to create sample w/ data should pass
assert client.register_sample(
name=samples[1],
dataset=dataset,
data_url='file://test/crud_samples/with-data.txt',
validate=False,
split='training')
listed_samples = client.list_samples(dataset)
assert len(listed_samples) == 2
# # Basic cleanup should work
client.delete_sample(samples[0], dataset)
client.delete_sample(samples[1], dataset)
assert len(client.list_samples(dataset)) == 0
client.delete_dataset(dataset)