12 lines
397 B
Python
12 lines
397 B
Python
import os
|
|
from vectorstore import create_vectorstore
|
|
|
|
def test_embeddings_type():
|
|
vectorstore = create_vectorstore()
|
|
embeddings = vectorstore.embedding_function
|
|
assert embeddings.__class__.__name__ == "OllamaEmbeddings"
|
|
assert embeddings.model == "nomic-embed-text"
|
|
|
|
def test_vectorstore_persistence():
|
|
vectorstore = create_vectorstore()
|
|
assert os.path.isdir("./chroma_db") |