refactor: Convert unittest assertions to pytest-style assert statements

This commit is contained in:
Jon Michael Aanes (aider) 2025-04-14 23:44:15 +02:00
parent a40a11b1b9
commit f37659e968

View File

@ -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)