Skip to content
Snippets Groups Projects
Unverified Commit 21dd2fdf authored by Siraj R Aizlewood's avatar Siraj R Aizlewood
Browse files

Linting.

parent bfeeb1d9
No related branches found
No related tags found
No related merge requests found
...@@ -89,7 +89,9 @@ class TestBaseLLM: ...@@ -89,7 +89,9 @@ class TestBaseLLM:
def test_mandatory_args_only(self, base_llm, mixed_function_schema): 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 assert base_llm._is_valid_inputs(
inputs, mixed_function_schema
) # True is implied
def test_all_args_provided(self, base_llm, mixed_function_schema): def test_all_args_provided(self, base_llm, mixed_function_schema):
inputs = { inputs = {
...@@ -98,7 +100,9 @@ class TestBaseLLM: ...@@ -98,7 +100,9 @@ class TestBaseLLM:
"optional1": "opt1", "optional1": "opt1",
"optional2": "opt2", "optional2": "opt2",
} }
assert base_llm._is_valid_inputs(inputs, mixed_function_schema) # True is implied assert base_llm._is_valid_inputs(
inputs, mixed_function_schema
) # True is implied
def test_missing_mandatory_arg(self, base_llm, mixed_function_schema): 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"}
...@@ -116,7 +120,9 @@ class TestBaseLLM: ...@@ -116,7 +120,9 @@ class TestBaseLLM:
def test_check_for_mandatory_inputs_all_present(self, base_llm, mandatory_params): def test_check_for_mandatory_inputs_all_present(self, base_llm, mandatory_params):
inputs = {"param1": "value1", "param2": "value2"} inputs = {"param1": "value1", "param2": "value2"}
assert base_llm._check_for_mandatory_inputs(inputs, mandatory_params) # True is implied assert base_llm._check_for_mandatory_inputs(
inputs, mandatory_params
) # True is implied
def test_check_for_mandatory_inputs_missing_one(self, base_llm, mandatory_params): def test_check_for_mandatory_inputs_missing_one(self, base_llm, mandatory_params):
inputs = {"param1": "value1"} inputs = {"param1": "value1"}
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment