a58398f5d4
- Cache persistant JSON thread-safe pour les résultats Ollama (invalidation par modèle) - Parallélisation des appels Ollama (ThreadPoolExecutor, 2 workers) - 6 nouvelles règles de filtrage DAS parasites (doublons, ponctuation, OCR, labo, fragments) - Client Ollama centralisé (mode JSON natif + retry) - Module GHM (estimation CMD/sévérité) - Module contrôle CPAM (parser + contre-argumentation RAG) - Export RUM (format RSS) - Viewer enrichi (détail dossier) Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
109 lines
3.7 KiB
Python
109 lines
3.7 KiB
Python
"""Tests unitaires pour le cache Ollama persistant."""
|
|
|
|
import json
|
|
import threading
|
|
|
|
import pytest
|
|
|
|
from src.medical.ollama_cache import OllamaCache
|
|
|
|
|
|
class TestOllamaCache:
|
|
def test_get_miss(self, tmp_path):
|
|
cache = OllamaCache(tmp_path / "cache.json", "gemma3:12b")
|
|
assert cache.get("HTA", "das") is None
|
|
|
|
def test_put_and_get(self, tmp_path):
|
|
cache = OllamaCache(tmp_path / "cache.json", "gemma3:12b")
|
|
result = {"code": "I10", "confidence": "high", "justification": "HTA essentielle"}
|
|
cache.put("HTA", "das", result)
|
|
assert cache.get("HTA", "das") == result
|
|
|
|
def test_key_normalization(self, tmp_path):
|
|
cache = OllamaCache(tmp_path / "cache.json", "gemma3:12b")
|
|
result = {"code": "I10", "confidence": "high"}
|
|
cache.put(" HTA ", "das", result)
|
|
assert cache.get("hta", "das") == result
|
|
|
|
def test_different_types_different_keys(self, tmp_path):
|
|
cache = OllamaCache(tmp_path / "cache.json", "gemma3:12b")
|
|
cache.put("Diabète", "dp", {"code": "E11.9"})
|
|
cache.put("Diabète", "das", {"code": "E11.8"})
|
|
assert cache.get("Diabète", "dp")["code"] == "E11.9"
|
|
assert cache.get("Diabète", "das")["code"] == "E11.8"
|
|
|
|
def test_save_and_reload(self, tmp_path):
|
|
path = tmp_path / "cache.json"
|
|
cache = OllamaCache(path, "gemma3:12b")
|
|
cache.put("HTA", "das", {"code": "I10"})
|
|
cache.save()
|
|
|
|
assert path.exists()
|
|
|
|
cache2 = OllamaCache(path, "gemma3:12b")
|
|
assert cache2.get("HTA", "das") == {"code": "I10"}
|
|
|
|
def test_save_no_write_if_clean(self, tmp_path):
|
|
path = tmp_path / "cache.json"
|
|
cache = OllamaCache(path, "gemma3:12b")
|
|
cache.save()
|
|
assert not path.exists()
|
|
|
|
def test_model_change_invalidates(self, tmp_path):
|
|
path = tmp_path / "cache.json"
|
|
cache = OllamaCache(path, "gemma3:12b")
|
|
cache.put("HTA", "das", {"code": "I10"})
|
|
cache.save()
|
|
|
|
cache2 = OllamaCache(path, "llama3:8b")
|
|
assert cache2.get("HTA", "das") is None
|
|
assert len(cache2) == 0
|
|
|
|
def test_corrupted_file(self, tmp_path):
|
|
path = tmp_path / "cache.json"
|
|
path.write_text("not valid json", encoding="utf-8")
|
|
|
|
cache = OllamaCache(path, "gemma3:12b")
|
|
assert len(cache) == 0
|
|
assert cache.get("HTA", "das") is None
|
|
|
|
def test_len(self, tmp_path):
|
|
cache = OllamaCache(tmp_path / "cache.json", "gemma3:12b")
|
|
assert len(cache) == 0
|
|
cache.put("HTA", "das", {"code": "I10"})
|
|
assert len(cache) == 1
|
|
cache.put("Diabète", "dp", {"code": "E11.9"})
|
|
assert len(cache) == 2
|
|
|
|
def test_thread_safety(self, tmp_path):
|
|
"""Écriture concurrente depuis plusieurs threads."""
|
|
cache = OllamaCache(tmp_path / "cache.json", "gemma3:12b")
|
|
errors = []
|
|
|
|
def writer(i):
|
|
try:
|
|
cache.put(f"diag_{i}", "das", {"code": f"X{i:02d}"})
|
|
except Exception as e:
|
|
errors.append(e)
|
|
|
|
threads = [threading.Thread(target=writer, args=(i,)) for i in range(20)]
|
|
for t in threads:
|
|
t.start()
|
|
for t in threads:
|
|
t.join()
|
|
|
|
assert not errors
|
|
assert len(cache) == 20
|
|
|
|
def test_json_format(self, tmp_path):
|
|
"""Le fichier JSON contient le modèle et les entrées."""
|
|
path = tmp_path / "cache.json"
|
|
cache = OllamaCache(path, "gemma3:12b")
|
|
cache.put("HTA", "das", {"code": "I10"})
|
|
cache.save()
|
|
|
|
raw = json.loads(path.read_text(encoding="utf-8"))
|
|
assert raw["model"] == "gemma3:12b"
|
|
assert "entries" in raw
|
|
assert len(raw["entries"]) == 1
|