From 71879695b951dd30bbd6cbd25261e22ac7f8e5d1 Mon Sep 17 00:00:00 2001
From: batsumaru <>
Date: Wed, 16 Sep 2026 17:33:42 +0900
Subject: Added more memory management and fixed system prompt bug

---
 tests/test_database.py | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

(limited to 'tests/test_database.py')

diff --git a/tests/test_database.py b/tests/test_database.py
index 290b985..e387123 100644
--- a/tests/test_database.py
+++ b/tests/test_database.py
@@ -34,7 +34,7 @@ def test_migrate_records_version(tmp_path):
     with database.connect() as connection:
         rows = connection.execute("SELECT version FROM schema_migrations").fetchall()
 
-    assert rows == [(1,)]
+    assert rows == [(1,), (2,), (3,)]
 
 
 def test_migrate_is_idempotent(tmp_path):
@@ -46,13 +46,13 @@ def test_migrate_is_idempotent(tmp_path):
     with database.connect() as connection:
         rows = connection.execute("SELECT version FROM schema_migrations").fetchall()
 
-    assert rows == [(1,)]
+    assert rows == [(1,), (2,), (3,)]
 
 
 def test_failed_migration_rolls_back(tmp_path):
     database = Database(tmp_path / "test.db")
 
-    migration = database.migrations_dir / "002_failing_migration.sql"
+    migration = database.migrations_dir / "999_failing_migration.sql"
     migration.write_text(
         """
         THIS IS INVALID SQL;
@@ -68,7 +68,7 @@ def test_failed_migration_rolls_back(tmp_path):
                 "SELECT version FROM schema_migrations"
             ).fetchall()
 
-        assert migrations == [(1,)]
+        assert migrations == [(1,), (2,), (3,)]
     finally:
         migration.unlink()
 
-- 
cgit v1.3