test(mediacloud): fix tests about time slicing

SocialChangeLab · Aug 4, 2024 · 3982555 · 3982555
1 parent cfd979b
commit 3982555
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/backend-python/media_impact_monitor/data_loaders/news_online/mediacloud_.py b/backend-python/media_impact_monitor/data_loaders/news_online/mediacloud_.py
@@ -105,7 +105,7 @@ def _story_list_all_pages(
 def _slice_date_range(start: date, end: date) -> list[tuple[date, date]]:
     result = []
     current = start.replace(day=1)
-    while current <= end:
+    while current <= min(end, date.today()):
         next_month = current + relativedelta(months=1)
         last_day = min(next_month - timedelta(days=1), date.today())
         result.append((current, last_day))

diff --git a/backend-python/media_impact_monitor/data_loaders/news_online/mediacloud_test.py b/backend-python/media_impact_monitor/data_loaders/news_online/mediacloud_test.py
@@ -36,7 +36,7 @@ def test_slicing_future_end_date():
 def test_slicing_same_month():
     start = date(2023, 3, 10)
     end = date(2023, 3, 20)
-    expected = [(date(2023, 3, 1), date(2023, 3, 20))]
+    expected = [(date(2023, 3, 1), date(2023, 3, 31))]
     assert _slice_date_range(start, end) == expected