Skip to content

Commit a88c145

Browse files
authored
update vector policy and vector query tests (#37647)
1 parent b7fe65c commit a88c145

File tree

4 files changed

+6
-4
lines changed

4 files changed

+6
-4
lines changed

sdk/cosmos/azure-cosmos/test/test_query_vector_similarity.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -137,7 +137,7 @@ def test_ordering_distances(self):
137137
"SimilarityScore FROM c ORDER BY VectorDistance(c.embedding, [{}], false, {{'distanceFunction': 'euclidean'}})" \
138138
.format(str(i), vector_string, vector_string)
139139

140-
flat_list = list(self.created_flat_euclidean_container.query_items(query=specs_query,
140+
flat_list = list(self.created_flat_euclidean_container.query_items(query=vanilla_query,
141141
enable_cross_partition_query=True))
142142
verify_ordering(flat_list, "euclidean")
143143

sdk/cosmos/azure-cosmos/test/test_query_vector_similarity_async.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -137,7 +137,7 @@ async def test_ordering_distances_async(self):
137137
"SimilarityScore FROM c ORDER BY VectorDistance(c.embedding, [{}], false, {{'distanceFunction': 'euclidean'}})" \
138138
.format(str(i), vector_string, vector_string)
139139

140-
flat_list = [item async for item in self.created_flat_euclidean_container.query_items(query=specs_query)]
140+
flat_list = [item async for item in self.created_flat_euclidean_container.query_items(query=vanilla_query)]
141141
verify_ordering(flat_list, "euclidean")
142142

143143
quantized_list = [item async for item in self.created_quantized_cosine_container.query_items(query=specs_query)]

sdk/cosmos/azure-cosmos/test/test_vector_policy.py

+2-1
Original file line numberDiff line numberDiff line change
@@ -163,7 +163,8 @@ def test_fail_replace_vector_indexing_policy(self):
163163
pytest.fail("Container replace should have failed for indexing policy.")
164164
except exceptions.CosmosHttpResponseError as e:
165165
assert e.status_code == 400
166-
assert "Vector Indexing Policy cannot be changed in Collection Replace" in e.http_error_message
166+
assert "Paths in existing vector indexing policy cannot be modified in Collection Replace." \
167+
" They can only be added or removed." in e.http_error_message
167168
self.test_db.delete_container(container_id)
168169

169170
def test_fail_create_vector_embedding_policy(self):

sdk/cosmos/azure-cosmos/test/test_vector_policy_async.py

+2-1
Original file line numberDiff line numberDiff line change
@@ -173,7 +173,8 @@ async def test_fail_replace_vector_indexing_policy_async(self):
173173
pytest.fail("Container replace should have failed for indexing policy.")
174174
except exceptions.CosmosHttpResponseError as e:
175175
assert e.status_code == 400
176-
assert "Vector Indexing Policy cannot be changed in Collection Replace" in e.http_error_message
176+
assert "Paths in existing vector indexing policy cannot be modified in Collection Replace." \
177+
" They can only be added or removed." in e.http_error_message
177178
await self.test_db.delete_container(container_id)
178179

179180
async def test_fail_create_vector_embedding_policy_async(self):

0 commit comments

Comments
 (0)