test/test_seen_issues_db.py

@@ -1,17 +1,18 @@
+import unittest
 
 from aider_gitea.seen_issues_db import SeenIssuesDB
 
 
-class TestSeenIssuesDB:
+class TestSeenIssuesDB(unittest.TestCase):
-    def setup_method(self):
+    def setUp(self):
         self.db = SeenIssuesDB(':memory:')
 
     def test_mark_and_check_seen_issue(self):
         issue_text = 'Test issue'
-        assert not self.db.has_seen(issue_text)
+        self.assertFalse(self.db.has_seen(issue_text))
         self.db.mark_as_seen(issue_text)
-        assert self.db.has_seen(issue_text)
+        self.assertTrue(self.db.has_seen(issue_text))
 
     def test_unseen_issue(self):
         issue_text = 'Unseen issue'
-        assert not self.db.has_seen(issue_text)
+        self.assertFalse(self.db.has_seen(issue_text))