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

Update tests

Browse source
This commit is contained in:
Cyber MacGeddon 2026-02-23 22:01:20 +00:00
parent 6bccfd3d55
commit d8dea2e8a8
1 changed files with 19 additions and 9 deletions
Download patch file Download diff file Expand all files Collapse all files

12
tests/unit/test_cli/test_tool_commands.py
View file

@ -406,7 +406,7 @@ class TestShowToolsStructuredQuery:
@patch('trustgraph.cli.show_tools.Api') @patch('trustgraph.cli.show_tools.Api')
def test_show_mixed_tool_types(self, mock_api_class, mock_api, capsys): def test_show_mixed_tool_types(self, mock_api_class, mock_api, capsys):
"""Test displaying multiple tool types including structured-query.""" """Test displaying multiple tool types including structured-query and row-embeddings-query."""
mock_api_class.return_value, mock_config = mock_api mock_api_class.return_value, mock_config = mock_api
tools = [ tools = [
@ -422,6 +422,13 @@ class TestShowToolsStructuredQuery:
"type": "structured-query", "type": "structured-query",
"collection": "sales" "collection": "sales"
}, },
{
"name": "find_customer",
"description": "Find customers by semantic search",
"type": "row-embeddings-query",
"schema-name": "customers",
"collection": "crm"
},
{ {
"name": "complete_text", "name": "complete_text",
"description": "Generate text", "description": "Generate text",
@ -443,11 +450,14 @@ class TestShowToolsStructuredQuery:
# All tool types should be displayed # All tool types should be displayed
assert "knowledge-query" in output assert "knowledge-query" in output
assert "structured-query" in output assert "structured-query" in output
assert "row-embeddings-query" in output
assert "text-completion" in output assert "text-completion" in output
# Collections should be shown for appropriate tools # Collections should be shown for appropriate tools
assert "docs" in output # knowledge-query collection assert "docs" in output # knowledge-query collection
assert "sales" in output # structured-query collection assert "sales" in output # structured-query collection
assert "crm" in output # row-embeddings-query collection
assert "customers" in output # row-embeddings-query schema-name
def test_show_main_parses_args_correctly(self): def test_show_main_parses_args_correctly(self):
"""Test that show main() parses arguments correctly.""" """Test that show main() parses arguments correctly."""