mirror of
https://github.com/crewAIInc/crewAI.git
synced 2026-01-08 23:58:34 +00:00
25 lines
1013 B
Python
25 lines
1013 B
Python
import sys
|
|
import unittest
|
|
from typing import Any, Dict, List, Optional
|
|
from unittest.mock import MagicMock, patch
|
|
|
|
import pytest
|
|
|
|
|
|
class TestOptionalChromadb(unittest.TestCase):
|
|
def test_rag_storage_import_error(self):
|
|
"""Test that RAGStorage raises an ImportError when chromadb is not installed."""
|
|
with patch.dict(sys.modules, {"chromadb": None}):
|
|
with pytest.raises(ImportError) as excinfo:
|
|
from crewai.memory.storage.rag_storage import RAGStorage
|
|
|
|
assert "ChromaDB is not installed" in str(excinfo.value)
|
|
|
|
def test_knowledge_storage_import_error(self):
|
|
"""Test that KnowledgeStorage raises an ImportError when chromadb is not installed."""
|
|
with patch.dict(sys.modules, {"chromadb": None}):
|
|
with pytest.raises(ImportError) as excinfo:
|
|
from crewai.knowledge.storage.knowledge_storage import KnowledgeStorage
|
|
|
|
assert "ChromaDB is not installed" in str(excinfo.value)
|