Fix tests

Florents-Tselai · Jun 23, 2024 · 0a3e01c · 0a3e01c
1 parent 5ad4e00
commit 0a3e01c
Show file tree

Hide file tree

Showing 3 changed files with 2 additions and 3 deletions.
diff --git a/setup.py b/setup.py
@@ -1,7 +1,7 @@
 from setuptools import setup
 import os
 
-VERSION = "0.0.0a0"
+VERSION = "0.0.1a0"
 
 
 def get_long_description():

diff --git a/tests/conftest.py b/tests/conftest.py
@@ -32,7 +32,7 @@ def existing_db_path(fresh_db_path):
     db = Database(fresh_db_path)
     table = db.create_table(
         "prompts",
-        {"prompt": str},
+        {"prompt": str, "generated": str}
     )
 
     table.insert({"prompt": "hello world!"})

diff --git a/tests/test_tsellm.py b/tests/test_tsellm.py
@@ -22,4 +22,3 @@ def test_cli_prompt_mock(existing_db_path):
         prompt_words = prompt.split()
         for word in words:
             assert word in prompt_words
-    assert existing_db_path == 'gdfg'