ed/manual_slop
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

test(rag): Fix test_rag_engine.py mock assertion

Browse Source 
Updated the test to correctly mock _get_sentence_transformers instead of SentenceTransformer to account for lazy loading imports.
This commit is contained in:
Edward R. Gonzalez
2026-05-14 20:24:23 -04:00
parent d7fab05f4a
commit 33e532a491
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/test_rag_engine.py
+2 -1
View File
@@ -31,7 +31,8 @@ def test_rag_engine_chroma(mock_chroma, mock_embed):
vs_config = models.VectorStoreConfig(provider='chroma', collection_name='test') vs_config = models.VectorStoreConfig(provider='chroma', collection_name='test')
config = models.RAGConfig(enabled=True, vector_store=vs_config, embedding_provider='local') config = models.RAGConfig(enabled=True, vector_store=vs_config, embedding_provider='local')
with patch('src.rag_engine.SentenceTransformer') as mock_st: with patch('src.rag_engine._get_sentence_transformers') as mock_st:
mock_st.return_value = MagicMock()
engine = RAGEngine(config) engine = RAGEngine(config)
assert engine.collection == mock_collection assert engine.collection == mock_collection