From b040c168c7d825085c23896dbe519fd94f54da73 Mon Sep 17 00:00:00 2001
From: Siraj R Aizlewood <siraj@aurelio.ai>
Date: Mon, 13 May 2024 21:30:20 +0400
Subject: [PATCH] Fixed pytests following resolution of conflicts with main.
---
tests/unit/llms/test_llm_base.py | 22 +++++++++++-----------
1 file changed, 11 insertions(+), 11 deletions(-)
diff --git a/tests/unit/llms/test_llm_base.py b/tests/unit/llms/test_llm_base.py
index 38d2ee73..a52a3415 100644
--- a/tests/unit/llms/test_llm_base.py
+++ b/tests/unit/llms/test_llm_base.py
@@ -11,11 +11,11 @@ class TestBaseLLM:
@pytest.fixture
def mixed_function_schema(self):
- return {
+ return [{
"name": "test_function",
"description": "A test function with mixed mandatory and optional parameters.",
"signature": "(mandatory1, mandatory2: int, optional1=None, optional2: str = 'default')",
- }
+ }]
@pytest.fixture
def mandatory_params(self):
@@ -90,34 +90,34 @@ class TestBaseLLM:
base_llm.extract_function_inputs(test_schema, test_query)
def test_mandatory_args_only(self, base_llm, mixed_function_schema):
- inputs = {"mandatory1": "value1", "mandatory2": 42}
+ inputs = [{"mandatory1": "value1", "mandatory2": 42}]
assert base_llm._is_valid_inputs(
inputs, mixed_function_schema
) # True is implied
def test_all_args_provided(self, base_llm, mixed_function_schema):
- inputs = {
+ inputs = [{
"mandatory1": "value1",
"mandatory2": 42,
"optional1": "opt1",
"optional2": "opt2",
- }
+ }]
assert base_llm._is_valid_inputs(
inputs, mixed_function_schema
) # True is implied
def test_missing_mandatory_arg(self, base_llm, mixed_function_schema):
- inputs = {"mandatory1": "value1", "optional1": "opt1", "optional2": "opt2"}
+ inputs = [{"mandatory1": "value1", "optional1": "opt1", "optional2": "opt2"}]
assert not base_llm._is_valid_inputs(inputs, mixed_function_schema)
def test_extra_arg_provided(self, base_llm, mixed_function_schema):
- inputs = {
+ inputs = [{
"mandatory1": "value1",
"mandatory2": 42,
"optional1": "opt1",
"optional2": "opt2",
"extra": "value",
- }
+ }]
assert not base_llm._is_valid_inputs(inputs, mixed_function_schema)
def test_check_for_mandatory_inputs_all_present(self, base_llm, mandatory_params):
@@ -201,11 +201,11 @@ class TestBaseLLM:
mocked_logger = mocker.patch("semantic_router.utils.logger.logger.error")
# Prepare inputs and a malformed function schema
- test_inputs = {"timezone": "America/New_York"}
+ test_inputs = {"timezone": "America/New_York"}
malformed_function_schema = {
"name": "get_time",
"description": "Finds the current time in a specific timezone.",
- "signature": "(timezone str)", # Malformed signature missing colon
+ "signiture": "(timezone: str)", # Malformed key name
"output": "<class 'str'>",
}
@@ -218,7 +218,7 @@ class TestBaseLLM:
assert not result, "Method should return False when an exception occurs"
# Check that the appropriate error message was logged
- expected_error_message = "Single input validation error: list index out of range" # Adjust based on the actual exception message
+ expected_error_message = "Single input validation error: 'signature'" # Adjust based on the actual exception message
mocked_logger.assert_called_once_with(expected_error_message)
def test_extract_parameter_info_valid(self, base_llm):
--
GitLab