apunkt/trustgraph
1
0
Fork 0
mirror of https://github.com/trustgraph-ai/trustgraph.git synced 2026-04-26 00:46:22 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

Merge 2.2 into master (#771)

Browse source
This commit is contained in:
cybermaggedon 2026-04-08 14:17:18 +01:00 • committed by GitHub
parent dc72ed3cca
commit befe951b2e
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
15 changed files with 2840 additions and 32 deletions
Download patch file Download diff file Expand all files Collapse all files

15
tests/unit/test_retrieval/test_graph_rag.py
View file

@ -465,12 +465,15 @@ class TestQuery:
return_value=(["entity1", "entity2"], ["concept1"])
)
query.follow_edges_batch = AsyncMock(return_value={
("entity1", "predicate1", "object1"),
("entity2", "predicate2", "object2")
})
query.follow_edges_batch = AsyncMock(return_value=(
{
("entity1", "predicate1", "object1"),
("entity2", "predicate2", "object2")
},
{}
))
subgraph, entities, concepts = await query.get_subgraph("test query")
subgraph, term_map, entities, concepts = await query.get_subgraph("test query")
query.get_entities.assert_called_once_with("test query")
query.follow_edges_batch.assert_called_once_with(["entity1", "entity2"], 1)
@ -503,7 +506,7 @@ class TestQuery:
test_entities = ["entity1", "entity3"]
test_concepts = ["concept1"]
query.get_subgraph = AsyncMock(
return_value=(test_subgraph, test_entities, test_concepts)
return_value=(test_subgraph, {}, test_entities, test_concepts)
)
async def mock_maybe_label(entity):