def test_datasets_iter(client):
keys = [n for n in range(10)]
client.publish_dataset(**{str(key): key for key in keys})
for n, key in enumerate(client.datasets):
assert key == str(n)
with pytest.raises(TypeError):
client.datasets.__aiter__()
def test_datasets_getitem(client):
for key in ["key", ("key", "key"), 1]:
value = "value"
client.publish_dataset(value, name=key)
assert client.datasets[key] == value
assert client.datasets.get(key) == value
assert client.datasets.get(key, default="something else") == value
def test_unpublish_sync(client):
data = client.scatter([0, 1, 2])
client.publish_dataset(data=data)
client.unpublish_dataset(name="data")
with pytest.raises(KeyError) as exc_info:
client.get_dataset(name="data")
assert "not found" in str(exc_info.value)
assert "data" in str(exc_info.value)
def test_unpublish_sync(client):
data = client.scatter([0, 1, 2])
client.publish_dataset(data=data)
client.unpublish_dataset(name='data')
with pytest.raises(KeyError) as exc_info:
result = client.get_dataset(name='data')
assert "not found" in str(exc_info.value)
assert "data" in str(exc_info.value)
def test_unpublish_multiple_datasets_sync(client):
x = delayed(inc)(1)
y = delayed(inc)(2)
client.publish_dataset(x=x, y=y)
client.unpublish_dataset(name="x")
with pytest.raises(KeyError) as exc_info:
client.get_dataset(name="x")
datasets = client.list_datasets()
assert set(datasets) == {"y"}
assert "not found" in str(exc_info.value)
assert "x" in str(exc_info.value)
client.unpublish_dataset(name="y")
with pytest.raises(KeyError) as exc_info:
client.get_dataset(name="y")
assert "not found" in str(exc_info.value)
assert "y" in str(exc_info.value)
def test_unpublish_multiple_datasets_sync(client):
x = delayed(inc)(1)
y = delayed(inc)(2)
client.publish_dataset(x=x, y=y)
client.unpublish_dataset(name='x')
with pytest.raises(KeyError) as exc_info:
result = client.get_dataset(name='x')
datasets = client.list_datasets()
assert set(datasets) == {'y'}
assert "not found" in str(exc_info.value)
assert "x" in str(exc_info.value)
client.unpublish_dataset(name='y')
with pytest.raises(KeyError) as exc_info:
result = client.get_dataset(name='y')
assert "not found" in str(exc_info.value)
assert "y" in str(exc_info.value)
def test_datasets_republish(client):
key, value, value2 = "key", "value", "value2"
client.publish_dataset(key=value)
assert client.get_dataset(key) == value
with pytest.raises(KeyError) as exc_info:
client.publish_dataset(key=value)
client.publish_dataset(key=value2, override=True)
assert client.get_dataset(key) == value2
def test_datasets_delitem(client):
for key in ["key", ("key", "key"), 1]:
value = "value"
client.publish_dataset(value, name=key)
del client.datasets[key]
assert key not in client.list_datasets()
def test_datasets_iter(client):
keys = [n for n in range(10)]
client.publish_dataset(**{str(key): key for key in keys})
for n, key in enumerate(client.datasets):
assert key == str(n)
def test_datasets_delitem(client):
for key in ['key', ('key', 'key'), 1]:
value = 'value'
client.publish_dataset(value, name=key)
del client.datasets[key]
assert key not in client.list_datasets()
def test_datasets_delitem(client):
for key in ['key', ('key', 'key'), 1]:
value = 'value'
client.publish_dataset(value, name=key)
del client.datasets[key]
assert key not in client.list_datasets()
def test_datasets_getitem(client):
for key in ['key', ('key', 'key'), 1]:
value = 'value'
client.publish_dataset(value, name=key)
assert client.datasets[key] == value
def test_datasets_iter(client):
keys = [n for n in range(10)]
client.publish_dataset(**{str(key): key for key in keys})
for n, key in enumerate(client.datasets):
assert key == str(n)
def test_datasets_getitem(client):
for key in ["key", ("key", "key"), 1]:
value = "value"
client.publish_dataset(value, name=key)
assert client.datasets[key] == value
def test_datasets_contains(client):
key, value = 'key', 'value'
client.publish_dataset(key=value)
assert key in client.datasets
def test_datasets_keys(client):
client.publish_dataset(**{str(n): n for n in range(10)})
keys = list(client.datasets.keys())
assert keys == [str(n) for n in range(10)]
def test_datasets_contains(client):
key, value = "key", "value"
client.publish_dataset(key=value)
assert key in client.datasets
def test_datasets_keys(client):
client.publish_dataset(**{str(n): n for n in range(10)})
keys = list(client.datasets.keys())
assert keys == [str(n) for n in range(10)]
def test_datasets_contains(client):
key, value = 'key', 'value'
client.publish_dataset(key=value)
assert key in client.datasets
def test_datasets_getitem(client):
for key in ['key', ('key', 'key'), 1]:
value = 'value'
client.publish_dataset(value, name=key)
assert client.datasets[key] == value