diff --git a/test/test_seen_issues_db.py b/test/test_seen_issues_db.py index da0bdd0..8096e11 100644 --- a/test/test_seen_issues_db.py +++ b/test/test_seen_issues_db.py @@ -1,18 +1,18 @@ -import unittest +import pytest from aider_gitea.seen_issues_db import SeenIssuesDB -class TestSeenIssuesDB(unittest.TestCase): - def setUp(self): +class TestSeenIssuesDB: + def setup_method(self): self.db = SeenIssuesDB(':memory:') def test_mark_and_check_seen_issue(self): issue_text = 'Test issue' - self.assertFalse(self.db.has_seen(issue_text)) + assert not self.db.has_seen(issue_text) self.db.mark_as_seen(issue_text) - self.assertTrue(self.db.has_seen(issue_text)) + assert self.db.has_seen(issue_text) def test_unseen_issue(self): issue_text = 'Unseen issue' - self.assertFalse(self.db.has_seen(issue_text)) + assert not self.db.has_seen(issue_text)