def test_delete(tmp_path, key, data, serializer):
store = CacheStore(str(tmp_path))
store.put(key=key, value=data, serializer=serializer)
assert key in store.keys()
store.delete(key=key, dry_run=True)
del store[key]
assert key not in store.keys()
def test_put_and_get(tmp_path, key, data, serializer):
store = CacheStore(str(tmp_path))
store.put(key=key, value=data, serializer=serializer)
assert key in store.keys()
assert isinstance(store.get_artifact(key), Artifact)
results = store[key]
if isinstance(data, (xr.Dataset, xr.DataArray)):
xr.testing.assert_equal(results, data)
elif isinstance(data, pd.DataFrame):
pd.testing.assert_frame_equal(results, data)
else:
assert results == data