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)
def test_register_sample_validate(client: PlatformClient):
dataset = 'blueno::test_register_sample_validate'
client.create_dataset(dataset)
with pytest.raises(PlatformError):
client.register_sample(
'test-sample',
dataset,
data_url='gs://elvo-platform/test/register_validate/no-data.xzx',
validate=True,
split='training')
assert len(client.list_samples(dataset)) == 0