Johan LEROY
314e3b72c0
`load_from_csv` gardait un `astype(bool)` sur `is_working_hours`, joué avant `_typer` : une case vide du CSV arrivait en `NaN` et en ressortait `True`, soit une heure ouvrée inventée. Le chemin base était corrigé, pas celui-ci, et rien ne le couvrait. La ligne disparaît, et `_typer` ramène désormais les colonnes de `FLAG_COLUMNS` à `float64` quel que soit le contenu lu : sans cela le dtype dépendait de l'écriture du fichier (`0`/`1` contre `True`/`False`) et de la présence d'un trou, et l'égalité de schéma entre les deux chargeurs que promet ML-START n'était vraie que par accident du jeu de test. `Seuils.seuil_biais` valait `0` et `_verdict` exigeait `> 0` : la règle était inerte partout, CLI et DAG compris, et aucun test ne l'exerçait. Elle reste désactivée par défaut, parce qu'un seuil en kWh ne se transpose pas d'un bureau de 10 kWh à une usine de 1 000 kWh et qu'aucune valeur n'a été calibrée sur la vraie série, mais `--bias-threshold` la rend atteignable et l'ADR 0011 porte l'arbitrage. Trois tests couvrent le chemin : inerte par défaut, dérive au-delà du seuil réglé, et priorité de la MAE sur le biais. Deux lignes de doc devenues fausses au passage : la signature de `load_recent_from_database` dans ML-START, qui omettait `until` devenu obligatoire, et la ligne `bias` de 20-backend, qui laissait croire que la métrique décide du verdict.
117 lines
3.5 KiB
Python
117 lines
3.5 KiB
Python
from datetime import UTC, datetime, timedelta
|
|
|
|
import pytest
|
|
|
|
from app.monitoring import drift as cli
|
|
from app.repositories.drift import NouveauRapportDerive
|
|
from app.services.drift import STATUT_DERIVE, STATUT_STABLE, Seuils
|
|
|
|
INSTANT = datetime(2026, 9, 22, 12, tzinfo=UTC)
|
|
|
|
|
|
def rapport(*, site_id: str | None, status: str, reason: str | None = None) -> NouveauRapportDerive:
|
|
return NouveauRapportDerive(
|
|
site_id=site_id,
|
|
window_start=INSTANT - timedelta(hours=168),
|
|
window_end=INSTANT,
|
|
reference_start=None,
|
|
reference_end=None,
|
|
n_observations=48,
|
|
mae=1.5,
|
|
mape=12.0,
|
|
bias=0.3,
|
|
reference_mae=1.2,
|
|
coverage_ratio=1.0,
|
|
insufficient_data_ratio=0.0,
|
|
model_references=["lightgbm-aaa"],
|
|
status=status,
|
|
reason=reason,
|
|
)
|
|
|
|
|
|
def installe(monkeypatch: pytest.MonkeyPatch, rapports: list[NouveauRapportDerive]) -> None:
|
|
async def fausse_execution(
|
|
*, now: datetime | None, site_id: str | None, seuils: Seuils | None
|
|
) -> list[NouveauRapportDerive]:
|
|
return rapports
|
|
|
|
monkeypatch.setattr(cli, "run_drift", fausse_execution)
|
|
|
|
|
|
def test_parse_args_defaults_to_the_standard_window() -> None:
|
|
arguments = cli.parse_args([])
|
|
|
|
assert arguments.window_hours == 168
|
|
assert arguments.grace_hours == 2
|
|
assert arguments.fail_on_drift is False
|
|
|
|
|
|
def test_parse_args_reads_the_site_id() -> None:
|
|
assert cli.parse_args(["--site-id", "SITE001"]).site_id == "SITE001"
|
|
|
|
|
|
def test_parse_args_parses_the_instant_option() -> None:
|
|
arguments = cli.parse_args(["--now", "2026-09-22T12:00:00+00:00"])
|
|
|
|
assert arguments.now == INSTANT
|
|
|
|
|
|
def test_parse_instant_treats_a_naive_datetime_as_utc() -> None:
|
|
assert cli._parse_instant("2026-09-22T12:00:00") == INSTANT
|
|
|
|
|
|
def test_seuils_depuis_translates_the_hour_options_into_durations() -> None:
|
|
seuils = cli.seuils_depuis(cli.parse_args(["--window-hours", "24", "--grace-hours", "1"]))
|
|
|
|
assert seuils.fenetre == timedelta(hours=24)
|
|
assert seuils.grace == timedelta(hours=1)
|
|
|
|
|
|
def test_main_prints_the_verdict_of_every_line(
|
|
monkeypatch: pytest.MonkeyPatch, capsys: pytest.CaptureFixture[str]
|
|
) -> None:
|
|
installe(
|
|
monkeypatch,
|
|
[
|
|
rapport(site_id="SITE001", status=STATUT_STABLE),
|
|
rapport(site_id=None, status=STATUT_STABLE),
|
|
],
|
|
)
|
|
|
|
code = cli.main([])
|
|
|
|
sortie = capsys.readouterr().out
|
|
assert code == 0
|
|
assert "SITE001" in sortie
|
|
assert "TOUS SITES" in sortie
|
|
|
|
|
|
def test_main_exits_non_zero_when_drift_is_detected_and_the_flag_is_set(
|
|
monkeypatch: pytest.MonkeyPatch, capsys: pytest.CaptureFixture[str]
|
|
) -> None:
|
|
installe(monkeypatch, [rapport(site_id=None, status=STATUT_DERIVE, reason="MAE doublée")])
|
|
|
|
code = cli.main(["--fail-on-drift"])
|
|
|
|
assert code == 1
|
|
assert "MAE doublée" in capsys.readouterr().out
|
|
|
|
|
|
def test_main_exits_zero_when_drift_is_detected_without_the_flag(
|
|
monkeypatch: pytest.MonkeyPatch, capsys: pytest.CaptureFixture[str]
|
|
) -> None:
|
|
installe(monkeypatch, [rapport(site_id=None, status=STATUT_DERIVE, reason="MAE doublée")])
|
|
|
|
code = cli.main([])
|
|
|
|
assert code == 0
|
|
assert capsys.readouterr().out != ""
|
|
|
|
|
|
def test_parse_args_leaves_the_bias_threshold_disabled_by_default() -> None:
|
|
assert cli.parse_args([]).bias_threshold == 0.0
|
|
|
|
|
|
def test_seuils_depuis_carries_the_bias_threshold() -> None:
|
|
assert cli.seuils_depuis(cli.parse_args(["--bias-threshold", "2.5"])).seuil_biais == 2.5
|