fix test

Gaudy Blanco · Gaudy Blanco · commit 7dc7c73370eb · 2025-10-15T23:41:59.000-06:00
diff --git a/tests/integration/vector_stores/test_azure_ai_search.py b/tests/integration/vector_stores/test_azure_ai_search.py
@@ -120,6 +120,7 @@ async def test_vector_store_operations(
             "vector": [0.1, 0.2, 0.3, 0.4, 0.5],
         }
 
+        vector_store.create_index()
         vector_store.load_documents(sample_documents)
         assert mock_index_client.create_or_update_index.called
         assert mock_search_client.upload_documents.called
@@ -188,6 +189,7 @@ async def test_vector_store_customization(
             vector_store_custom.vector_field: [0.1, 0.2, 0.3, 0.4, 0.5],
         }
 
+        vector_store_custom.create_index()
         vector_store_custom.load_documents(sample_documents)
         assert mock_index_client.create_or_update_index.called
         assert mock_search_client.upload_documents.called

Original file line number	Diff line number	Diff line change
`@@ -120,6 +120,7 @@ async def test_vector_store_operations(`
`120`	`120`	`"vector": [0.1, 0.2, 0.3, 0.4, 0.5],`
`121`	`121`	`}`
`122`	`122`
	`123`	`+ vector_store.create_index()`
`123`	`124`	`vector_store.load_documents(sample_documents)`
`124`	`125`	`assert mock_index_client.create_or_update_index.called`
`125`	`126`	`assert mock_search_client.upload_documents.called`
`@@ -188,6 +189,7 @@ async def test_vector_store_customization(`
`188`	`189`	`vector_store_custom.vector_field: [0.1, 0.2, 0.3, 0.4, 0.5],`
`189`	`190`	`}`
`190`	`191`
	`192`	`+ vector_store_custom.create_index()`
`191`	`193`	`vector_store_custom.load_documents(sample_documents)`
`192`	`194`	`assert mock_index_client.create_or_update_index.called`
`193`	`195`	`assert mock_search_client.upload_documents.called`