Create configured model cache directories
This commit is contained in:
parent
3b7584e1b1
commit
98efb8cb78
1 changed files with 23 additions and 1 deletions
22
tests/large_file/test_storage_safety.py
Normal file
22
tests/large_file/test_storage_safety.py
Normal file
|
|
@ -0,0 +1,22 @@
|
|||
from pathlib import Path
|
||||
from typing import Iterator
|
||||
|
||||
import pytest
|
||||
from great_ai.large_file import LargeFileBase, LargeFileLocal
|
||||
|
||||
|
||||
@pytest.fixture(autouse=True)
|
||||
def storage(tmp_path: Path, monkeypatch: pytest.MonkeyPatch) -> Iterator[None]:
|
||||
for implementation in (LargeFileBase, LargeFileLocal):
|
||||
monkeypatch.setattr(implementation, "cache_path", tmp_path / "cache")
|
||||
monkeypatch.setattr(implementation, "max_cache_size", None)
|
||||
yield
|
||||
|
||||
|
||||
def test_configured_cache_directory_is_created(
|
||||
tmp_path: Path, monkeypatch: pytest.MonkeyPatch
|
||||
) -> None:
|
||||
monkeypatch.setattr(LargeFileLocal, "cache_path", tmp_path / "custom-cache")
|
||||
with LargeFileLocal("model", "w") as file:
|
||||
file.write("weights")
|
||||
assert LargeFileLocal("model").get().read_text() == "weights"
|
||||
Loading…
Reference in a new issue