From 6d6ac11185f1fc77535eb2bd6cebba6932c0aba5 Mon Sep 17 00:00:00 2001 From: Sam Cox Date: Wed, 20 Dec 2023 15:04:20 -0800 Subject: [PATCH] added test for check args in skill --- tests/test_agent.py | 13 +++++++++++++ 1 file changed, 13 insertions(+) diff --git a/tests/test_agent.py b/tests/test_agent.py index aaf10200..e7a9c7c6 100644 --- a/tests/test_agent.py +++ b/tests/test_agent.py @@ -163,3 +163,16 @@ def test_execute_skill_function(skill_manager): with pytest.raises(ValueError) as excinfo: skill_manager.execute_skill_function("nonexistent_tool", path_registry) assert "Code for nonexistent_tool not found" in str(excinfo.value) + + +def test_check_arguments_success(skill_manager): + skill_manager.skills = { + "sample_tool": {"arguments": [{"name": "arg1"}, {"name": "arg2"}]} + } + try: + skill_manager._check_arguments("sample_tool", arg1=5, arg2=10) + except ValueError: + pytest.fail("ValueError raised unexpectedly") + with pytest.raises(ValueError) as excinfo: + skill_manager._check_arguments("sample_tool", arg1=5) + assert "Missing arguments" in str(excinfo.value)