def test_datasets_getitem_default(client):
with pytest.raises(KeyError) as exc_info:
client.get_dataset("key")
assert client.datasets.get("key", default="value") == "value"
assert client.datasets.get("key", default=None) is None
assert client.get_dataset("key", default="value") == "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_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_setitem(client):
for key in ["key", ("key", "key"), 1]:
value = "value"
client.datasets[key] = value
assert client.get_dataset(key) == value
assert client.get_dataset(key, default="something else") == value
def test_datasets_setitem(client):
for key in ['key', ('key', 'key'), 1]:
value = 'value'
client.datasets[key] = value
assert client.get_dataset(key) == value
def test_datasets_setitem(client):
for key in ["key", ("key", "key"), 1]:
value = "value"
client.datasets[key] = value
assert client.get_dataset(key) == value
def test_datasets_setitem(client):
for key in ['key', ('key', 'key'), 1]:
value = 'value'
client.datasets[key] = value
assert client.get_dataset(key) == value
