update

Author: thomas

tests/tests_data/cache/test_cache.py

@@ -21,6 +21,7 @@
 from lightning import seed_everything
 from lightning.data.datasets.env import _DistributedEnv
 from lightning.data.streaming import Cache
+from lightning.data.streaming import cache as cache_module
 from lightning.data.streaming.dataloader import StreamingDataLoader
 from lightning.fabric import Fabric
 from lightning.pytorch.demos.boring_classes import RandomDataset
@@ -203,3 +204,18 @@ def __len__(self) -> int:
     with pytest.raises(ValueError, match="Your dataset items aren't deterministic"):
         for batch in dataloader:
             pass
+
+
+def test_cache_with_name(tmpdir, monkeypatch):
+    with pytest.raises(FileNotFoundError, match="The provided cache directory"):
+        Cache(name="something")
+
+    os.makedirs(os.path.join(tmpdir, "something"), exist_ok=True)
+    os.makedirs(os.path.join(tmpdir, "remote_dir"), exist_ok=True)
+    monkeypatch.setattr(cache_module, "_try_create_cache_dir", lambda name: os.path.join(tmpdir, name))
+
+    monkeypatch.setattr(cache_module, "_find_remote_dir", lambda name, _: (os.path.join(tmpdir, "remote_dir"), True))
+    cache = Cache(name="something")
+    assert cache._writer._chunk_size == 2
+    assert cache._writer._cache_dir == os.path.join(tmpdir, "something")
+    assert cache._reader._remote_dir == os.path.join(tmpdir, "remote_dir")