'Test that cohort data is included in grades csv if cohort configuration is enabled for course.'

def test_cohort_data_in_grading(self):

    cohort_groups = ['cohort    1', 'cohort    2']
    course = CourseFactory.create(cohort_config={'cohorted': True, 'auto_cohort': True, 'auto_cohort_groups': cohort_groups})
    user_1 = 'user_1'
    user_2 = 'user_2'
    CourseEnrollment.enroll(UserFactory.create(username=user_1), course.id)
    CourseEnrollment.enroll(UserFactory.create(username=user_2), course.id)
    self._verify_cell_data_for_user(user_1, course.id, 'Cohort    Name', '')
    self._verify_cell_data_for_user(user_2, course.id, 'Cohort    Name', '')
