diff --git a/tests/congruence_tests/test_query.py b/tests/congruence_tests/test_query.py index 4e63e557..b09c8743 100644 --- a/tests/congruence_tests/test_query.py +++ b/tests/congruence_tests/test_query.py @@ -1657,6 +1657,31 @@ def test_query_group(): init_client(local_client, secondary_collection_points, SECONDARY_COLLECTION_NAME) init_client(http_client, secondary_collection_points, SECONDARY_COLLECTION_NAME) + http_client.create_payload_index( + COLLECTION_NAME, field_name="id", field_schema=models.PayloadSchemaType.INTEGER + ) + http_client.create_payload_index( + COLLECTION_NAME, field_name="rand_digit", field_schema=models.PayloadSchemaType.INTEGER + ) + http_client.create_payload_index( + COLLECTION_NAME, field_name="two_words", field_schema=models.PayloadSchemaType.KEYWORD + ) + http_client.create_payload_index( + COLLECTION_NAME, + field_name="city.name", + field_schema=models.PayloadSchemaType.KEYWORD, + ) + http_client.create_payload_index( + COLLECTION_NAME, + field_name="maybe", + field_schema=models.PayloadSchemaType.KEYWORD, + ) + http_client.create_payload_index( + COLLECTION_NAME, + field_name="maybe_null", + field_schema=models.PayloadSchemaType.KEYWORD, + ) + searcher.group_size = 5 searcher.limit = 3 for key in group_by_keys(): @@ -1683,6 +1708,7 @@ def test_query_group(): ) searcher.group_by = "city.name" + for i in range(100): query_filter = one_random_filter_please() try: