Skip to content

Commit a810c1c

Browse files
Gaudy BlancoGaudy Blanco
authored andcommitted
test_factory.py integration test fixes
1 parent a735808 commit a810c1c

File tree

1 file changed

+3
-6
lines changed

1 file changed

+3
-6
lines changed

tests/integration/vector_stores/test_factory.py

Lines changed: 3 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -18,12 +18,11 @@
1818

1919
def test_create_lancedb_vector_store():
2020
kwargs = {
21-
"collection_name": "test_collection",
2221
"db_uri": "/tmp/lancedb",
2322
}
2423
vector_store = VectorStoreFactory.create_vector_store(
2524
vector_store_type=VectorStoreType.LanceDB.value,
26-
vector_store_schema_config=VectorStoreSchemaConfig(),
25+
vector_store_schema_config=VectorStoreSchemaConfig(index_name="test_collection"),
2726
kwargs=kwargs,
2827
)
2928
assert isinstance(vector_store, LanceDBVectorStore)
@@ -33,13 +32,12 @@ def test_create_lancedb_vector_store():
3332
@pytest.mark.skip(reason="Azure AI Search requires credentials and setup")
3433
def test_create_azure_ai_search_vector_store():
3534
kwargs = {
36-
"collection_name": "test_collection",
3735
"url": "https://test.search.windows.net",
3836
"api_key": "test_key",
3937
}
4038
vector_store = VectorStoreFactory.create_vector_store(
4139
vector_store_type=VectorStoreType.AzureAISearch.value,
42-
vector_store_schema_config=VectorStoreSchemaConfig(),
40+
vector_store_schema_config=VectorStoreSchemaConfig(index_name="test_collection"),
4341
kwargs=kwargs,
4442
)
4543
assert isinstance(vector_store, AzureAISearchVectorStore)
@@ -48,14 +46,13 @@ def test_create_azure_ai_search_vector_store():
4846
@pytest.mark.skip(reason="CosmosDB requires credentials and setup")
4947
def test_create_cosmosdb_vector_store():
5048
kwargs = {
51-
"collection_name": "test_collection",
5249
"connection_string": "AccountEndpoint=https://test.documents.azure.com:443/;AccountKey=test_key==",
5350
"database_name": "test_db",
5451
}
5552

5653
vector_store = VectorStoreFactory.create_vector_store(
5754
vector_store_type=VectorStoreType.CosmosDB.value,
58-
vector_store_schema_config=VectorStoreSchemaConfig(),
55+
vector_store_schema_config=VectorStoreSchemaConfig(index_name="test_collection"),
5956
kwargs=kwargs,
6057
)
6158

0 commit comments

Comments
 (0)