apunkt/trustgraph
1
0
Fork 0
mirror of https://github.com/trustgraph-ai/trustgraph.git synced 2026-07-24 12:41:02 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions 2

More tests

Browse source
This commit is contained in:
Cyber MacGeddon 2025-09-04 16:00:55 +01:00
parent fc6cebf140
commit 13ca380be6
2 changed files with 5 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

6
tests/integration/test_nlp_query_integration.py
View file

@ -124,7 +124,7 @@ class TestNLPQueryServiceIntegration:
phase2_response = PromptResponse(
text=json.dumps({
"query": expected_graphql.strip(),
"variables": {"min_total": 500.0},
"variables": {"min_total": "500.0"},
"confidence": 0.92
}),
error=None
@ -153,7 +153,7 @@ class TestNLPQueryServiceIntegration:
assert "California" in response.graphql_query
assert response.detected_schemas == ["customers", "orders"]
assert response.confidence == 0.92
assert response.variables["min_total"] == 500.0
assert response.variables["min_total"] == "500.0"
@pytest.mark.asyncio
async def test_complex_multi_table_query_integration(self, integration_processor):
@ -444,7 +444,7 @@ class TestNLPQueryServiceIntegration:
msg.properties.return_value = {"id": f"concurrent-test-{i}"}
flow = MagicMock()
flow_response = MagicMock()
flow_response = AsyncMock()
flow.return_value = flow_response
requests.append(request)

4
tests/unit/test_retrieval/test_nlp_query.py
View file

@ -162,7 +162,7 @@ class TestNLPQueryProcessor:
consumer = MagicMock()
flow = MagicMock()
flow_response = MagicMock()
flow_response = AsyncMock()
flow.return_value = flow_response
# Mock Phase 1 response
@ -217,7 +217,7 @@ class TestNLPQueryProcessor:
consumer = MagicMock()
flow = MagicMock()
flow_response = MagicMock()
flow_response = AsyncMock()
flow.return_value = flow_response
# Mock Phase 1 error