Fix additional tests
This commit is contained in:
@@ -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)
|
||||
|
||||
Reference in New Issue
Block a user