From c6e9f85f60167e66fc4aa5617ed37d3382b730ba Mon Sep 17 00:00:00 2001
From: Siraj R Aizlewood <siraj@aurelio.ai>
Date: Mon, 13 May 2024 21:34:35 +0400
Subject: [PATCH] Linting.
---
tests/unit/llms/test_llm_base.py | 44 ++++++++++++++++++--------------
1 file changed, 25 insertions(+), 19 deletions(-)
diff --git a/tests/unit/llms/test_llm_base.py b/tests/unit/llms/test_llm_base.py
index a52a3415..3699ded0 100644
--- a/tests/unit/llms/test_llm_base.py
+++ b/tests/unit/llms/test_llm_base.py
@@ -11,11 +11,13 @@ class TestBaseLLM:
@pytest.fixture
def mixed_function_schema(self):
- return [{
- "name": "test_function",
- "description": "A test function with mixed mandatory and optional parameters.",
- "signature": "(mandatory1, mandatory2: int, optional1=None, optional2: str = 'default')",
- }]
+ 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):
@@ -96,12 +98,14 @@ class TestBaseLLM:
) # True is implied
def test_all_args_provided(self, base_llm, mixed_function_schema):
- inputs = [{
- "mandatory1": "value1",
- "mandatory2": 42,
- "optional1": "opt1",
- "optional2": "opt2",
- }]
+ inputs = [
+ {
+ "mandatory1": "value1",
+ "mandatory2": 42,
+ "optional1": "opt1",
+ "optional2": "opt2",
+ }
+ ]
assert base_llm._is_valid_inputs(
inputs, mixed_function_schema
) # True is implied
@@ -111,13 +115,15 @@ class TestBaseLLM:
assert not base_llm._is_valid_inputs(inputs, mixed_function_schema)
def test_extra_arg_provided(self, base_llm, mixed_function_schema):
- inputs = [{
- "mandatory1": "value1",
- "mandatory2": 42,
- "optional1": "opt1",
- "optional2": "opt2",
- "extra": "value",
- }]
+ 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,7 +207,7 @@ 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.",
--
GitLab