diff --git a/tests/unit/clients/test_enhance_client.py b/tests/unit/clients/test_enhance_client.py index 1db34c4..0097850 100644 --- a/tests/unit/clients/test_enhance_client.py +++ b/tests/unit/clients/test_enhance_client.py @@ -17,7 +17,7 @@ async def test_mocked_enhance(self, mocker, client): # ARRANGE # Define the search query parameters - params = {"query": "your_search_query", "limit": 10} + params = {"query": "your_search_query", "size": 10} # Mock the _post_or_put method mocker.patch.object( @@ -43,7 +43,7 @@ async def test_mocked_create_bulkjob(self, mocker, client): # ARRANGE # Define the search query parameters - params = {"query": "your_bulk_enhance_query", "limit": 10} + params = {"query": "your_bulk_enhance_query", "size": 10} # Mock the _post_or_put method mocker.patch.object( diff --git a/tests/unit/clients/test_search_client.py b/tests/unit/clients/test_search_client.py index aa055ae..9624a95 100644 --- a/tests/unit/clients/test_search_client.py +++ b/tests/unit/clients/test_search_client.py @@ -19,7 +19,7 @@ async def test_mocked_search(self, mocker, client): # ARRANGE # Define the search query parameters - params = {"query": "your_search_query", "limit": 10} + params = {"query": "your_search_query", "size": 10} # Mock the _post_or_put method mocker.patch.object(