"""Tests rapides : python -m pytest -q   (ou python tests/test_grapp.py)"""
import sys
from pathlib import Path

sys.path.insert(0, str(Path(__file__).resolve().parents[1]))
from grapp.carte import trouver_vins
from grapp.engine import DicoVino, Grapp, Preferences

G = Grapp(use_hf=False)
NOMS = G.vins["nom"].tolist()


def test_filtre_couleur_et_bulles():
    r = G.recommander(Preferences(couleur="blanc", bulles="oui"), k=10)
    assert r and all(x.vin["couleur"] == "blanc" and x.vin["bulles"] == 1 for x in r)


def test_huitres_donne_un_blanc_mineral():
    p = Preferences(couleur="blanc", familles=["mineral"], occasion="repas",
                    format="bouteille", budget=25, envie="huîtres et fruits de mer")
    assert G.recommander(p, k=1)[0].vin["famille"] == "mineral"


def test_lecture_carte_tolere_fautes_ocr():
    texte = "Cétes du Rhone, E. Guigal 2021 27€\nChampagne Brut, Maison X 68 €\nChablis ler Cru 48 €"
    res = {x["nom"]: x["prix_carte"] for x in trouver_vins(texte, NOMS)}
    assert res == {"Côtes du Rhône": 27.0, "Champagne Brut": 68.0, "Chablis": 48.0}


def test_reco_limitee_a_la_carte_et_prix_resto():
    r = G.recommander(Preferences(format="bouteille"), carte={"Chinon": 29.0, "Tavel": 30.0})
    assert {x.nom for x in r} <= {"Chinon", "Tavel"} and {x.prix for x in r} <= {29.0, 30.0}


def test_dico():
    assert "Tanins" in DicoVino(use_hf=False).chercher("tanins")["terme"].tolist()


if __name__ == "__main__":
    for n, f in list(globals().items()):
        if n.startswith("test_"):
            f(); print("OK", n)