Skip to content

Commit

Permalink
Indention fix
Browse files Browse the repository at this point in the history
  • Loading branch information
Florents-Tselai committed Jun 30, 2024
1 parent 3f38c63 commit c832668
Showing 1 changed file with 7 additions and 7 deletions.
14 changes: 7 additions & 7 deletions tests/test_tsellm.py
Original file line number Diff line number Diff line change
Expand Up @@ -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__":
Expand Down

0 comments on commit c832668

Please sign in to comment.