Files
creator2/tests/test_diagramme.py
2026-07-12 16:13:50 +02:00

131 lines
6.6 KiB
Python

"""Ebene 3.5 Diagramme: deterministische DAG-Diagramme (kein LLM), Spec→Mermaid."""
import db
import diagramme
import llm
from conftest import run_anlegen, topic_anlegen
def _baustein_mit_kette(topic, n=4):
"""Baustein mit n Atomen und einer braucht-Kette a0←a1←…←a(n-1) (n-1 Kanten)."""
soll = db.insert("soll", topic=topic, punkt="P", status="bestaetigt", belege=db.j([]))
ziel = db.insert("lernziele", topic=topic, text="Kann P", soll_id=soll, status="aktiv")
b = db.insert("bausteine", topic=topic, ziel_id=ziel, titel="B", ord=0, status="neu")
atome = [db.insert("atome", topic=topic, titel=f"Atom {i}", typ="begriff",
definition="d", status="neu", baustein_id=b, ord=i, braucht=db.j([]))
for i in range(n)]
for i in range(1, n): # Atom i braucht Atom i-1
db.execute("INSERT INTO kanten(topic, von_atom, zu_atom, art) VALUES(?,?,?,'braucht')",
(topic, atome[i], atome[i - 1]))
return b, atome
def test_spec_zu_mermaid_flowchart():
spec = {"typ": "dependency",
"knoten": [{"id": "1", "label": "A"}, {"id": "2", "label": 'B "x" [y]'}],
"kanten": [{"von": "1", "zu": "2", "label": ""}]}
out = diagramme._spec_zu_mermaid(spec)
assert out.startswith("flowchart TD")
assert 'n1["A"]' in out and "n1 --> n2" in out
assert '"' not in out.split("\n", 1)[1].replace('["', "").replace('"]', "") # Label entschärft
async def test_dag_diagramm_wird_gebaut_und_verifiziert():
topic = topic_anlegen("diag-dag")
b, atome = _baustein_mit_kette(topic, n=4) # 3 Kanten ≥ DIAGRAMM_MIN_KANTEN
ctx = llm.Kontext(run_anlegen(topic), topic, "minimax")
await diagramme.bauen(ctx)
d = db.one("SELECT * FROM diagramme WHERE baustein_id=?", (b,))
assert d and d["status"] == "verifiziert" and d["quelle"] == "dag"
assert d["mermaid"].startswith("flowchart TD")
assert all(f"n{a}" in d["mermaid"] for a in atome) # jeder Knoten drin
assert diagramme.messen(ctx) == [] # keine offenen Kandidaten
assert diagramme.gate(ctx) is None
async def test_wenig_kanten_kein_diagramm():
topic = topic_anlegen("diag-leer")
b, _ = _baustein_mit_kette(topic, n=2) # nur 1 Kante < Schwelle
ctx = llm.Kontext(run_anlegen(topic), topic, "minimax")
await diagramme.bauen(ctx)
# kein GEBAUTES Diagramm (Fake-Judge sagt noetig=False → nur eine 'kein'-Merkzeile)
assert db.query("SELECT id FROM diagramme WHERE topic=? AND status='verifiziert'",
(topic,)) == []
assert db.one("SELECT status FROM diagramme WHERE baustein_id=?", (b,))["status"] == "kein"
async def test_llm_diagramm_plan_spec_verify(monkeypatch):
"""LLM-Zweig: Judge sagt nötig → Spec → Mermaid → Grounding-Panel verifiziert."""
import fake_agents
topic = topic_anlegen("diag-llm")
# Baustein OHNE genug Kanten (kein DAG) → LLM-Zweig entscheidet
soll = db.insert("soll", topic=topic, punkt="P", status="bestaetigt", belege=db.j([]))
ziel = db.insert("lernziele", topic=topic, text="Kann P", soll_id=soll, status="aktiv")
b = db.insert("bausteine", topic=topic, ziel_id=ziel, titel="Automat", ord=0, status="neu")
for i in range(3):
db.insert("atome", topic=topic, titel=f"Zustand {i}", typ="begriff", definition="d",
status="neu", baustein_id=b, ord=i, braucht=db.j([]))
monkeypatch.setitem(fake_agents._HANDLER, "Diagramm-Plan",
lambda p: {"noetig": True, "typ": "state"})
ctx = llm.Kontext(run_anlegen(topic), topic, "minimax")
await diagramme.bauen(ctx)
d = db.one("SELECT * FROM diagramme WHERE baustein_id=?", (b,))
assert d["status"] == "verifiziert" and d["quelle"] == "llm" and d["typ"] == "state"
assert d["mermaid"].startswith("stateDiagram-v2")
async def test_llm_diagramm_grounding_verwirft(monkeypatch):
"""Grounding-Panel einstimmig ok=false → Diagramm verworfen (fail-closed)."""
import fake_agents
topic = topic_anlegen("diag-verwirf")
soll = db.insert("soll", topic=topic, punkt="P", status="bestaetigt", belege=db.j([]))
ziel = db.insert("lernziele", topic=topic, text="Kann P", soll_id=soll, status="aktiv")
b = db.insert("bausteine", topic=topic, ziel_id=ziel, titel="X", ord=0, status="neu")
for i in range(3):
db.insert("atome", topic=topic, titel=f"K{i}", typ="begriff", definition="d",
status="neu", baustein_id=b, ord=i, braucht=db.j([]))
monkeypatch.setitem(fake_agents._HANDLER, "Diagramm-Plan",
lambda p: {"noetig": True, "typ": "flow"})
monkeypatch.setitem(fake_agents._HANDLER, "Diagramm-Verify",
lambda p: [{"ok": False, "mangel": "erfundener Knoten"}])
ctx = llm.Kontext(run_anlegen(topic), topic, "minimax")
await diagramme.bauen(ctx)
assert db.one("SELECT status FROM diagramme WHERE baustein_id=?", (b,))["status"] == "verworfen"
async def test_marker_wird_zu_fence():
"""Assembly: platzierter <!-- diagramm: id --> → ```mermaid-Fence."""
import guide
topic = topic_anlegen("diag-place")
b, _ = _baustein_mit_kette(topic, n=4)
ctx = llm.Kontext(run_anlegen(topic), topic, "minimax")
await diagramme.bauen(ctx)
d = db.one("SELECT id FROM diagramme WHERE baustein_id=?", (b,))
db.insert("sections", baustein_id=b, stage="done", text_kompakt="- k",
text_lang=f"Text.\n\n<!-- diagramm: {d['id']} -->\n\nMehr Text.")
lang = guide.kapitel_struktur(topic)[0]["sections"][0]["lang"]
assert "```mermaid" in lang and "flowchart TD" in lang
assert "<!-- diagramm:" not in lang # Marker ersetzt
async def test_toter_diagramm_marker_wird_entfernt():
import guide
topic = topic_anlegen("diag-tot")
b, _ = _baustein_mit_kette(topic, n=2) # kein Diagramm gebaut
db.insert("sections", baustein_id=b, stage="done", text_kompakt="- k",
text_lang="Text.\n\n<!-- diagramm: 999 -->\n\nMehr.")
lang = guide.kapitel_struktur(topic)[0]["sections"][0]["lang"]
assert "<!-- diagramm:" not in lang and "```mermaid" not in lang
async def test_guide_messen_meldet_toten_marker():
"""Ein Marker ohne verifiziertes Diagramm → diagramm_marker_tot in guide.messen."""
import guide
topic = topic_anlegen("diag-qa")
b, _ = _baustein_mit_kette(topic, n=4)
db.insert("sections", baustein_id=b, stage="done", text_kompakt="- k",
text_lang="Text.\n\n<!-- diagramm: 777 -->\n\n" + "Wort " * 60)
ctx = llm.Kontext(run_anlegen(topic), topic, "minimax")
ctx.ebene = "guide"
assert "diagramm_marker_tot" in {x["art"] for x in guide.messen(ctx)}