def test_get_memberships(self, username, course_ids, team_ids, expected_count): self.assertEqual( CourseTeamMembership.get_memberships(username=username, course_ids=course_ids, team_ids=team_ids).count(), expected_count)
def load_course_team_memberships(self): """ Caches existing team memberships by (user_id, teamset_id) """ for membership in CourseTeamMembership.get_memberships(course_ids=[self.course.id]): user_id = membership.user_id teamset_id = membership.team.topic_id self.existing_course_team_memberships[(user_id, teamset_id)] = membership.team
def test_get_memberships(self, username, course_ids, team_id, expected_count): self.assertEqual( CourseTeamMembership.get_memberships(username=username, course_ids=course_ids, team_id=team_id).count(), expected_count )