From ae26c8b50dc1ad6ab6e92f9c333b5131b53caa02 Mon Sep 17 00:00:00 2001 From: PatrykWyzgowski Date: Fri, 19 Jul 2024 10:20:56 +0200 Subject: [PATCH] Pre-commit fixes. --- tests/unit/test_collection.py | 14 ++++++++++---- 1 file changed, 10 insertions(+), 4 deletions(-) diff --git a/tests/unit/test_collection.py b/tests/unit/test_collection.py index 38adf5f8..a077286d 100644 --- a/tests/unit/test_collection.py +++ b/tests/unit/test_collection.py @@ -13,8 +13,14 @@ from dbally.iql import IQLQuery from dbally.iql.syntax import FunctionCall from dbally.views.exposed_functions import ExposedFunction, MethodParamWithTyping -from tests.unit.mocks import MockIQLGenerator, MockLLM, MockSimilarityIndex, MockViewBase, MockViewSelector, \ - MockAggregationFormatter +from tests.unit.mocks import ( + MockAggregationFormatter, + MockIQLGenerator, + MockLLM, + MockSimilarityIndex, + MockViewBase, + MockViewSelector, +) class MockView1(MockViewBase): @@ -298,7 +304,7 @@ async def test_ask_view_selection_single_view() -> None: result = await collection.ask("Mock question") assert result.view_name == "MockViewWithResults" assert result.results == [{"foo": "bar"}] - assert result.context == {"baz": "qux", "iql": {'aggregation': 'test_aggregation()', 'filters': 'test_filter()'}} + assert result.context == {"baz": "qux", "iql": {"aggregation": "test_aggregation()", "filters": "test_filter()"}} async def test_ask_view_selection_multiple_views() -> None: @@ -319,7 +325,7 @@ async def test_ask_view_selection_multiple_views() -> None: result = await collection.ask("Mock question") assert result.view_name == "MockViewWithResults" assert result.results == [{"foo": "bar"}] - assert result.context == {"baz": "qux", "iql": {'aggregation': 'test_aggregation()', 'filters': 'test_filter()'}} + assert result.context == {"baz": "qux", "iql": {"aggregation": "test_aggregation()", "filters": "test_filter()"}} async def test_ask_view_selection_no_views() -> None: