def _get_forecasts(students, course):
"""Get the forecast dates for all the students.
If there are no students, a generic forecast is added to the None key.
"""
forecaster = Forecaster()
forecasts = {}
for student in students:
forecasts[student] = forecaster.get_items_by_task(student, course)
if not students:
forecasts[None] = forecaster.get_items_by_task(student=None,
course=course)
return forecasts
def test_no_student_forecast(self):
"""The forecaster can produce a forecast for no student."""
enrollment = EnrollmentFactory()
course = CourseFactory(grade_levels=[enrollment.grade_level])
task = CourseTaskFactory(course=course)
forecaster = Forecaster()
items = forecaster.get_items_by_task(None, course)
assert items[task]["planned_date"] == datetime.date.today()
