mirror of
https://github.com/arc53/DocsGPT
synced 2024-11-17 21:26:26 +00:00
20 lines
716 B
Python
20 lines
716 B
Python
"""
|
|
Tests regarding the vector store class, including checking
|
|
compatibility between different transformers and local vector
|
|
stores (index.faiss)
|
|
"""
|
|
import pytest
|
|
from application.vectorstore.faiss import FaissStore
|
|
from application.core.settings import settings
|
|
|
|
def test_init_local_faiss_store_huggingface():
|
|
"""
|
|
Test that asserts that trying to initialize a FaissStore with
|
|
the huggingface sentence transformer below together with the
|
|
index.faiss file in the application/ folder results in a
|
|
dimension mismatch error.
|
|
"""
|
|
settings.EMBEDDINGS_NAME = "huggingface_sentence-transformers/all-mpnet-base-v2"
|
|
with pytest.raises(ValueError):
|
|
FaissStore("application/", "", None)
|