Skip to content

Commit

Permalink
Move vector search query test methods into separate class
Browse files Browse the repository at this point in the history
Maximum number of public method is reached, hence,
move vector serach query test methods into
separate class.

Signed-off-by: Vijayan Balasubramanian <[email protected]>
  • Loading branch information
VijayanB committed Dec 25, 2023
1 parent d4add27 commit 9f07636
Showing 1 changed file with 3 additions and 2 deletions.
5 changes: 3 additions & 2 deletions tests/worker_coordinator/runner_test.py
Original file line number Diff line number Diff line change
Expand Up @@ -2207,6 +2207,8 @@ async def test_search_pipeline_using_request_params(self, opensearch):
)
opensearch.clear_scroll.assert_not_called()


class VectorSearchQueryRunnerTests(TestCase):
@mock.patch("opensearchpy.OpenSearch")
@run_async
async def test_query_vector_search_with_perfect_recall(self, opensearch):
Expand Down Expand Up @@ -2338,7 +2340,6 @@ async def test_query_vector_search_with_no_results(self, opensearch):
headers={"Accept-Encoding": "identity"}
)


@mock.patch("opensearchpy.OpenSearch")
@run_async
async def test_query_vector_search_with_imperfect_recall(self, opensearch):
Expand Down Expand Up @@ -2559,7 +2560,6 @@ async def test_query_vector_search_with_zero_recall_1(self, opensearch):
headers={"Accept-Encoding": "identity"}
)


@mock.patch("opensearchpy.OpenSearch")
@run_async
async def test_query_vector_search_with_custom_id_field(self, opensearch):
Expand Down Expand Up @@ -2736,6 +2736,7 @@ async def test_query_vector_search_with_custom_id_field_inside_source(self, open
headers={"Accept-Encoding": "identity"}
)


class CreateIngestPipelineRunnerTests(TestCase):
@mock.patch("opensearchpy.OpenSearch")
@run_async
Expand Down

0 comments on commit 9f07636

Please sign in to comment.