From c832668f5ef71c5731dfb58eb480d0810756ff68 Mon Sep 17 00:00:00 2001 From: Florents Tselai Date: Sun, 30 Jun 2024 21:37:03 +0300 Subject: [PATCH] Indention fix --- tests/test_tsellm.py | 14 +++++++------- 1 file changed, 7 insertions(+), 7 deletions(-) diff --git a/tests/test_tsellm.py b/tests/test_tsellm.py index 3005e0a..51a1f51 100644 --- a/tests/test_tsellm.py +++ b/tests/test_tsellm.py @@ -100,13 +100,13 @@ def test_embed_hazo_binary(self): self.expect_success(":memory:", "select embed(randomblob(16), 'hazo')") -def test_embed_default_hazo(self): - self.assertEqual(llm_cli.get_default_embedding_model(), "hazo") - out = self.expect_success(":memory:", "select embed('hello world')") - self.assertEqual( - "('[5.0, 5.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0]',)\n", - out, - ) + def test_embed_default_hazo(self): + self.assertEqual(llm_cli.get_default_embedding_model(), "hazo") + out = self.expect_success(":memory:", "select embed('hello world')") + self.assertEqual( + "('[5.0, 5.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0]',)\n", + out, + ) if __name__ == "__main__":