fix: test_vectorstore.py — Экзамен: RAG-агент с ChromaDB и веб-поиском

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")