diff --git a/test_vectorstore.py b/test_vectorstore.py new file mode 100644 index 0000000..6475a83 --- /dev/null +++ b/test_vectorstore.py @@ -0,0 +1,12 @@ +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") \ No newline at end of file