test: fix calendar program tests

program/tests/test_calendar_program.py

@@ -21,7 +21,7 @@ def url(include_virtual=False, start=None, end=None):
 def assert_episodes(episodes, one_week=False) -> None:
     """asserts the episodes are valid."""
 
-    assert len(episodes) == 1 if not one_week else 7
+    assert len(episodes) == 2 if not one_week else 7
 
     for episode in episodes:
         assert episode["id"]
@@ -31,20 +31,24 @@ def assert_episodes(episodes, one_week=False) -> None:
 def assert_program(program, show, fallback_show=None, one_week=False) -> None:
     """asserts the program are valid and correspond to the given show and fallback show."""
 
-    assert len(program) == 1 if fallback_show is None and not one_week else (7 if one_week else 3)
+    assert len(program) == 2 if fallback_show is None and not one_week else (7 if one_week else 3)
 
-    assert program[0]["end"]
-    assert program[0]["id"]
-    assert program[0]["start"]
+    for entry in program:
+        assert entry["end"]
+        assert entry["id"]
+        assert entry["start"]
 
     if fallback_show is None:
-        assert program[0]["showId"] == show.id
-        assert program[0]["timeslotId"]
+        for entry in program:
+            assert entry["showId"] == show.id
+            assert entry["timeslotId"]
     else:
-        assert program[0]["showId"] == fallback_show.id
-
-        assert program[1]["showId"] == show.id
-        assert program[1]["timeslotId"]
+        for entry in program:
+            if entry["timeslotId"]:
+                assert entry["showId"] == show.id
+                assert entry["timeslotId"]
+            else:
+                assert entry["showId"] == fallback_show.id
 
 
 def assert_shows(shows, show, fallback_show=None) -> None:
@@ -66,7 +70,7 @@ def assert_shows(shows, show, fallback_show=None) -> None:
 def assert_timeslots(timeslots, show, one_week=False) -> None:
     """asserts the timeslots are valid and correspond to the given show."""
 
-    assert len(timeslots) == 1 if not one_week else 7
+    assert len(timeslots) == 2 if not one_week else 7
 
     for timeslot in timeslots:
         assert timeslot["id"]