wehub-resource-sync
66 lines
2.0 KiB
Python
66 lines
2.0 KiB
Python
"""Regression tests for cross-platform cache encoding."""
|
|
|
|
from __future__ import annotations
|
|
|
|
import json
|
|
import time
|
|
|
|
import whichllm.models.benchmark as benchmark_mod
|
|
import whichllm.models.cache as cache_mod
|
|
|
|
|
|
class _ReadableCacheFile:
|
|
def __init__(self, payload: dict):
|
|
self.payload = payload
|
|
self.encoding = None
|
|
|
|
def exists(self) -> bool:
|
|
return True
|
|
|
|
def read_text(self, *, encoding: str | None = None) -> str:
|
|
self.encoding = encoding
|
|
return json.dumps(self.payload, ensure_ascii=False)
|
|
|
|
|
|
class _WritableCacheFile:
|
|
def __init__(self):
|
|
self.encoding = None
|
|
self.text = None
|
|
|
|
def write_text(self, text: str, *, encoding: str | None = None) -> int:
|
|
self.encoding = encoding
|
|
self.text = text
|
|
return len(text)
|
|
|
|
|
|
def test_model_cache_reads_and_writes_utf8(monkeypatch, tmp_path):
|
|
reader = _ReadableCacheFile(
|
|
{"cached_at": time.time(), "models": [{"id": "test/Omega-Ω"}]}
|
|
)
|
|
monkeypatch.setattr(cache_mod, "CACHE_FILE", reader)
|
|
assert cache_mod.load_cache() == [{"id": "test/Omega-Ω"}]
|
|
assert reader.encoding == "utf-8"
|
|
|
|
writer = _WritableCacheFile()
|
|
monkeypatch.setattr(cache_mod, "CACHE_DIR", tmp_path)
|
|
monkeypatch.setattr(cache_mod, "CACHE_FILE", writer)
|
|
cache_mod.save_cache([{"id": "test/Omega-Ω"}])
|
|
assert writer.encoding == "utf-8"
|
|
assert "Ω" in writer.text
|
|
|
|
|
|
def test_benchmark_cache_reads_and_writes_utf8(monkeypatch, tmp_path):
|
|
reader = _ReadableCacheFile(
|
|
{"cached_at": time.time(), "scores": {"test/Omega-Ω": 1.0}}
|
|
)
|
|
monkeypatch.setattr(benchmark_mod, "BENCHMARK_CACHE", reader)
|
|
assert benchmark_mod.load_benchmark_cache() == {"test/Omega-Ω": 1.0}
|
|
assert reader.encoding == "utf-8"
|
|
|
|
writer = _WritableCacheFile()
|
|
monkeypatch.setattr(benchmark_mod, "CACHE_DIR", tmp_path)
|
|
monkeypatch.setattr(benchmark_mod, "BENCHMARK_CACHE", writer)
|
|
benchmark_mod.save_benchmark_cache({"test/Omega-Ω": 1.0})
|
|
assert writer.encoding == "utf-8"
|
|
assert "Ω" in writer.text
|