andras/great-ai
andras/great-ai
Watch
1
0
Fork
You've already forked great-ai
 0
Code Issues Pull requests 2 Projects Releases 4 Packages 1 Wiki Activity Actions

Count model directory contents against the cache quota

Browse source
This commit is contained in:
Andras Schmelczer 2026-10-27 16:52:35 +00:00
commit fb31b3ed83
1 changed files with 17 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

11
tests/large_file/test_storage_safety.py
View file

@ -45,3 +45,14 @@ def test_cached_directories_can_be_removed(
monkeypatch.setattr(LargeFileLocal, "max_cache_size", "0")
reader.clean_up()
assert not cached.exists()
def test_cache_quota_counts_directory_contents(
tmp_path: Path, monkeypatch: pytest.MonkeyPatch
) -> None:
source = tmp_path / "source"
source.mkdir()
(source / "weights").write_bytes(b"x" * 100_000)
monkeypatch.setattr(LargeFileLocal, "max_cache_size", "1 KB")
LargeFileLocal("model", "w").push(source)
assert not (LargeFileLocal.cache_path / "model-0").exists()