update vector policy and vector query tests (#37647)

Azure · Sep 30, 2024 · a88c145 · a88c145
1 parent b7fe65c
commit a88c145
Show file tree

Hide file tree

Showing 4 changed files with 6 additions and 4 deletions.
diff --git a/sdk/cosmos/azure-cosmos/test/test_query_vector_similarity.py b/sdk/cosmos/azure-cosmos/test/test_query_vector_similarity.py
@@ -137,7 +137,7 @@ def test_ordering_distances(self):
                           "SimilarityScore FROM c ORDER BY VectorDistance(c.embedding, [{}], false, {{'distanceFunction': 'euclidean'}})" \
                 .format(str(i), vector_string, vector_string)
 
-            flat_list = list(self.created_flat_euclidean_container.query_items(query=specs_query,
+            flat_list = list(self.created_flat_euclidean_container.query_items(query=vanilla_query,
                                                                                enable_cross_partition_query=True))
             verify_ordering(flat_list, "euclidean")
 

diff --git a/sdk/cosmos/azure-cosmos/test/test_query_vector_similarity_async.py b/sdk/cosmos/azure-cosmos/test/test_query_vector_similarity_async.py
@@ -137,7 +137,7 @@ async def test_ordering_distances_async(self):
                           "SimilarityScore FROM c ORDER BY VectorDistance(c.embedding, [{}], false, {{'distanceFunction': 'euclidean'}})" \
                 .format(str(i), vector_string, vector_string)
 
-            flat_list = [item async for item in self.created_flat_euclidean_container.query_items(query=specs_query)]
+            flat_list = [item async for item in self.created_flat_euclidean_container.query_items(query=vanilla_query)]
             verify_ordering(flat_list, "euclidean")
 
             quantized_list = [item async for item in self.created_quantized_cosine_container.query_items(query=specs_query)]

diff --git a/sdk/cosmos/azure-cosmos/test/test_vector_policy.py b/sdk/cosmos/azure-cosmos/test/test_vector_policy.py
@@ -163,7 +163,8 @@ def test_fail_replace_vector_indexing_policy(self):
             pytest.fail("Container replace should have failed for indexing policy.")
         except exceptions.CosmosHttpResponseError as e:
             assert e.status_code == 400
-            assert "Vector Indexing Policy cannot be changed in Collection Replace" in e.http_error_message
+            assert "Paths in existing vector indexing policy cannot be modified in Collection Replace." \
+                   " They can only be added or removed." in e.http_error_message
         self.test_db.delete_container(container_id)
 
     def test_fail_create_vector_embedding_policy(self):

diff --git a/sdk/cosmos/azure-cosmos/test/test_vector_policy_async.py b/sdk/cosmos/azure-cosmos/test/test_vector_policy_async.py
@@ -173,7 +173,8 @@ async def test_fail_replace_vector_indexing_policy_async(self):
             pytest.fail("Container replace should have failed for indexing policy.")
         except exceptions.CosmosHttpResponseError as e:
             assert e.status_code == 400
-            assert "Vector Indexing Policy cannot be changed in Collection Replace" in e.http_error_message
+            assert "Paths in existing vector indexing policy cannot be modified in Collection Replace." \
+                   " They can only be added or removed." in e.http_error_message
         await self.test_db.delete_container(container_id)
 
     async def test_fail_create_vector_embedding_policy_async(self):