From 94a7fe5d4d6dcc33c9a8fb3ad42a1a15cd891e62 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Luk=C3=A1=C5=A1=20Kucharczyk?= Date: Tue, 12 May 2026 11:04:04 +0200 Subject: [PATCH] Fix additional tests --- tests/test_session_formatting.py | 6 ++++-- tests/test_streak.py | 13 ++++++------- 2 files changed, 10 insertions(+), 9 deletions(-) diff --git a/tests/test_session_formatting.py b/tests/test_session_formatting.py index 4faacfc..8607773 100644 --- a/tests/test_session_formatting.py +++ b/tests/test_session_formatting.py @@ -22,11 +22,13 @@ class FormatDurationTest(TestCase): g = Game(name="The Test Game") g.save() p = Purchase( - game=g, date_purchased=datetime(2022, 9, 26, 14, 58, tzinfo=ZONEINFO) + date_purchased=datetime(2022, 9, 26, 14, 58, tzinfo=ZONEINFO) ) p.save() + p.games.add(g) + p.save() s = Session( - purchase=p, + game=g, timestamp_start=datetime(2022, 9, 26, 14, 58, tzinfo=ZONEINFO), timestamp_end=datetime(2022, 9, 26, 17, 38, tzinfo=ZONEINFO), ) diff --git a/tests/test_streak.py b/tests/test_streak.py index 8162798..8ee5b90 100644 --- a/tests/test_streak.py +++ b/tests/test_streak.py @@ -5,7 +5,6 @@ from common.time import daterange, streak_bruteforce class StreakTest(unittest.TestCase): - streak = streak_bruteforce def test_daterange_exclusive(self): d = daterange(date(2024, 8, 1), date(2024, 8, 3)) @@ -22,14 +21,14 @@ class StreakTest(unittest.TestCase): ) def test_1day_streak(self): - self.assertEqual(streak([date(2024, 8, 1)])["days"], 1) + self.assertEqual(streak_bruteforce([date(2024, 8, 1)])["days"], 1) def test_2day_streak(self): - self.assertEqual(streak([date(2024, 8, 1), date(2024, 8, 2)])["days"], 2) + self.assertEqual(streak_bruteforce([date(2024, 8, 1), date(2024, 8, 2)])["days"], 2) def test_31day_streak(self): self.assertEqual( - streak(daterange(date(2024, 8, 1), date(2024, 8, 31), end_inclusive=True))[ + streak_bruteforce(daterange(date(2024, 8, 1), date(2024, 8, 31), end_inclusive=True))[ "days" ], 31, @@ -39,14 +38,14 @@ class StreakTest(unittest.TestCase): d = daterange( date(2024, 8, 1), date(2024, 8, 5), end_inclusive=True ) + daterange(date(2024, 8, 7), date(2024, 8, 10), end_inclusive=True) - self.assertEqual(streak(d)["days"], 5) + self.assertEqual(streak_bruteforce(d)["days"], 5) def test_10day_streak_in_31_days(self): d = daterange(date(2024, 8, 1), date(2024, 8, 31), end_inclusive=True) d.remove(date(2024, 8, 8)) d.remove(date(2024, 8, 15)) d.remove(date(2024, 8, 21)) - self.assertEqual(streak(d)["days"], 10) + self.assertEqual(streak_bruteforce(d)["days"], 10) def test_10day_streak_in_31_days_with_consecutive_missing(self): d = daterange(date(2024, 8, 1), date(2024, 8, 31), end_inclusive=True) @@ -57,4 +56,4 @@ class StreakTest(unittest.TestCase): d.remove(date(2024, 8, 8)) d.remove(date(2024, 8, 15)) d.remove(date(2024, 8, 21)) - self.assertEqual(streak(d)["days"], 10) + self.assertEqual(streak_bruteforce(d)["days"], 10)