Test cli

.readthedocs.yml

@@ -3,7 +3,7 @@ version: 2
 build:
   os: ubuntu-22.04
   tools:
-    python: "3.11"
+    python: "3.12"
 
 sphinx:
   configuration: docs/conf.py

tests/test_tsellm.py

@@ -1,12 +1,9 @@
 from sqlite_utils import Database
-
+import llm
 from tsellm.cli import cli
-import pytest
-import datetime
-from click.testing import CliRunner
 
 
-def test_cli(db_path):
+def test_tsellm_cli(db_path):
     db = Database(db_path)
     assert [] == db.table_names()
     table = db.create_table(
@@ -28,3 +25,10 @@ def test_cli(db_path):
         ("hello",),
         ("world",),
     ]
+
+    cli([db_path, "UPDATE prompts SET generated=prompt(prompt)"])
+
+    assert db.execute("select prompt, generated from prompts").fetchall() == [
+        ("hello", "hellohello"),
+        ("world", "worldworld"),
+    ]

tsellm/cli.py

@@ -126,5 +126,3 @@ def cli(*args):
             console.interact(banner, exitmsg="")
     finally:
         con.close()
-
-    sys.exit(0)