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