diff --git a/tests/embeder_test.py b/tests/embedder_test.py similarity index 75% rename from tests/embeder_test.py rename to tests/embedder_test.py index 214b9110..d64dae11 100644 --- a/tests/embeder_test.py +++ b/tests/embedder_test.py @@ -9,25 +9,26 @@ def MockConfig(): 'debug_mode': False, 'continuous_mode': False, 'speak_mode': False, - 'memory_embeder': 'sbert' + 'memory_embedder': 'sbert' }) -class TestMemoryEmbeder(unittest.TestCase): - def setUp(self): +class TestMemoryEmbedder(unittest.TestCase): + def setUp(self): self.cfg = MockConfig() def test_ada(self): - self.cfg.memory_embeder = "ada" + self.cfg.memory_embedder = "ada" text = "Sample text" result = get_embedding(text) self.assertEqual(result.shape, (1536,)) def test_sbert(self): - self.cfg.memory_embeder = "sbert" + self.cfg.memory_embedder = "sbert" text = "Sample text" result = get_embedding(text) self.assertEqual(result.shape, (768,)) if __name__ == '__main__': - unittest.main() \ No newline at end of file + unittest.main() + \ No newline at end of file