102 lines
2.7 KiB
Python
102 lines
2.7 KiB
Python
#!/usr/bin/env python3
|
|
"""Tests für Second Brain Kern-Module."""
|
|
|
|
import sys
|
|
import os
|
|
import tempfile
|
|
sys.path.insert(0, os.path.join(os.path.dirname(__file__), ".."))
|
|
|
|
from src.engram import Engram, Grounding, Correctness
|
|
from src.store import EngramStore
|
|
from src.retriever import Retriever
|
|
|
|
|
|
def test_engram_creation():
|
|
eg = Engram.create("Test content", source="test", tags=["a", "b"])
|
|
assert eg.content == "Test content"
|
|
assert eg.metadata["source"] == "test"
|
|
assert "a" in eg.metadata["tags"]
|
|
assert eg.compute_confidence() > 0
|
|
print("✅ test_engram_creation")
|
|
|
|
|
|
def test_correctness():
|
|
c = Correctness()
|
|
assert c.score() == 0.5
|
|
c.confirm("user")
|
|
assert c.score() == 1.0
|
|
c.reject("user")
|
|
assert c.score() == 0.5
|
|
print("✅ test_correctness")
|
|
|
|
|
|
def test_store_crud():
|
|
with tempfile.NamedTemporaryFile(suffix=".sqlite", delete=False) as f:
|
|
db_path = f.name
|
|
|
|
store = EngramStore(db_path)
|
|
eg = Engram.create("Store test", tags=["store"])
|
|
store.save(eg)
|
|
|
|
loaded = store.get(str(eg.id))
|
|
assert loaded is not None
|
|
assert loaded.content == "Store test"
|
|
assert store.count() == 1
|
|
|
|
store.delete(str(eg.id))
|
|
assert store.count() == 0
|
|
|
|
store.close()
|
|
os.unlink(db_path)
|
|
print("✅ test_store_crud")
|
|
|
|
|
|
def test_search():
|
|
with tempfile.NamedTemporaryFile(suffix=".sqlite", delete=False) as f:
|
|
db_path = f.name
|
|
|
|
store = EngramStore(db_path)
|
|
store.save(Engram.create("Python ist eine Programmiersprache", tags=["coding"]))
|
|
store.save(Engram.create("SQLite ist eine Datenbank", tags=["db"]))
|
|
store.save(Engram.create("JavaScript ist eine Sprache", tags=["coding"]))
|
|
|
|
ret = Retriever(store)
|
|
results = ret.retrieve("Programmiersprache Datenbank")
|
|
assert len(results) >= 2
|
|
|
|
results = ret.retrieve("nichtsexistenterbegriff12345")
|
|
assert len(results) == 0
|
|
|
|
store.close()
|
|
os.unlink(db_path)
|
|
print("✅ test_search")
|
|
|
|
|
|
def test_links():
|
|
eg1 = Engram.create("Parent")
|
|
eg2 = Engram.create("Child")
|
|
eg1.add_link(eg2)
|
|
assert eg2.id in eg1.links
|
|
assert eg1.id in eg2.links
|
|
print("✅ test_links")
|
|
|
|
|
|
def test_grounding():
|
|
eg = Engram.create("Test", grounding=Grounding.VERIFIED)
|
|
assert eg.metadata["grounding"] == 4
|
|
conf = eg.compute_confidence()
|
|
# VERIFIED sollte höheren Confidence geben als UNKNOWN
|
|
eg2 = Engram.create("Test2", grounding=Grounding.UNKNOWN)
|
|
assert eg.compute_confidence() > eg2.compute_confidence()
|
|
print("✅ test_grounding")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
test_engram_creation()
|
|
test_correctness()
|
|
test_store_crud()
|
|
test_search()
|
|
test_links()
|
|
test_grounding()
|
|
print("\n🎉 Alle 6 Tests bestanden!")
|