def test_get_summary_and_trend_data(self, mock_enrollment): mock_enrollment.return_value = utils.get_mock_api_enrollment_data(self.course_id) actual_summary, actual_trend = self.presenter.get_summary_and_trend_data() self.assertDictEqual(actual_summary, utils.get_mock_enrollment_summary()) expected_trend = utils.get_mock_presenter_enrollment_trend(self.course_id) self.assertListEqual(actual_trend, expected_trend)
def test_hide_empty_enrollment_modes(self, mock_enrollment): """ Enrollment modes with no enrolled students should not be returned. """ mock_enrollment.return_value = utils.get_mock_api_enrollment_data(self.course_id, include_verified=False) actual_summary, actual_trend = self.presenter.get_summary_and_trend_data() self.assertDictEqual(actual_summary, utils.get_mock_enrollment_summary(include_verified=False)) expected_trend = utils.get_mock_presenter_enrollment_trend(self.course_id, include_verified=False) self.assertListEqual(actual_trend, expected_trend)