(self)
| 13 | |
| 14 | class TestSimpleStructMemReader(unittest.TestCase): |
| 15 | def setUp(self): |
| 16 | # Mock config |
| 17 | self.config = MagicMock(spec=SimpleStructMemReaderConfig) |
| 18 | self.config.llm = MagicMock() |
| 19 | self.config.general_llm = None # Optional, falls back to main llm |
| 20 | self.config.embedder = MagicMock() |
| 21 | self.config.chunker = MagicMock() |
| 22 | self.config.remove_prompt_example = MagicMock() |
| 23 | |
| 24 | # Mock dependencies |
| 25 | with ( |
| 26 | patch.object(LLMFactory, "from_config", return_value=MagicMock()), |
| 27 | patch.object(EmbedderFactory, "from_config", return_value=MagicMock()), |
| 28 | patch.object(ChunkerFactory, "from_config", return_value=MagicMock()), |
| 29 | ): |
| 30 | self.reader = SimpleStructMemReader(self.config) |
| 31 | |
| 32 | # Set up mock LLM and embedder |
| 33 | self.reader.llm = MagicMock() |
| 34 | self.reader.general_llm = self.reader.llm # Falls back to main llm |
| 35 | self.reader.embedder = MagicMock() |
| 36 | self.reader.chunker = MagicMock() |
| 37 | |
| 38 | def test_init(self): |
| 39 | """Test initialization of the reader.""" |
nothing calls this directly
no test coverage detected