15 lines
447 B
Python
15 lines
447 B
Python
import pytest
|
|
from genesis.services.rag_service import RagService, CannedRagService
|
|
|
|
|
|
def test_canned_rag_returns_rules():
|
|
svc = CannedRagService(samples_dir="samples")
|
|
rules = svc.retrieve_write_rules("db_design")
|
|
assert isinstance(rules, list)
|
|
rules2 = svc.retrieve_design_rules("db_design")
|
|
assert isinstance(rules2, list)
|
|
|
|
|
|
def test_rag_service_is_protocol():
|
|
assert isinstance(CannedRagService("samples"), RagService)
|