From 34eac5754c2d9dfba18b9b89407400ce2da19e71 Mon Sep 17 00:00:00 2001 From: Tymec Date: Fri, 14 Apr 2023 18:06:47 +0200 Subject: [PATCH] test: fix typo and add newline at the end - Fixed "embeder" typo to "embedder" - Added newline at the end of test unit --- tests/{embeder_test.py => embedder_test.py} | 13 +++++++------ 1 file changed, 7 insertions(+), 6 deletions(-) rename tests/{embeder_test.py => embedder_test.py} (75%) 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