mirror of
https://github.com/deepset-ai/haystack.git
synced 2025-07-16 05:20:51 +00:00
4fa9d2d8e7
* fix milvus and faiss tests not running * fix schema manually * fix test_dpr_embedding test for milvus * pip freeze on milvus tests * fix milvus1 tests being executed: fix all_doc_stores order * Revert "pip freeze on milvus tests" This reverts commit 75ebb6f7e507bb8477e87d9e63b4a294f7946cab. * make infer_required_doc_store more robust * don't skip tests without docstore requirements * use markers for docstore tests
25 lines
1.2 KiB
Python
25 lines
1.2 KiB
Python
def pytest_addoption(parser):
|
|
parser.addoption(
|
|
"--document_store_type",
|
|
action="store",
|
|
default="elasticsearch, faiss, sql, memory, milvus1, milvus, weaviate, pinecone, opensearch",
|
|
)
|
|
|
|
|
|
def pytest_generate_tests(metafunc):
|
|
# Get selected docstores from CLI arg
|
|
document_store_type = metafunc.config.option.document_store_type
|
|
selected_doc_stores = [item.strip() for item in document_store_type.split(",")]
|
|
|
|
# parametrize document_store fixture if it's in the test function argument list
|
|
# but does not have an explicit parametrize annotation e.g
|
|
# @pytest.mark.parametrize("document_store", ["memory"], indirect=False)
|
|
found_mark_parametrize_document_store = False
|
|
for marker in metafunc.definition.iter_markers("parametrize"):
|
|
if "document_store" in marker.args[0]:
|
|
found_mark_parametrize_document_store = True
|
|
break
|
|
# for all others that don't have explicit parametrization, we add the ones from the CLI arg
|
|
if "document_store" in metafunc.fixturenames and not found_mark_parametrize_document_store:
|
|
metafunc.parametrize("document_store", selected_doc_stores, indirect=True)
|