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