def test_lookup_team_membership_data(self): with self.assertNumQueries(3): # pylint: disable=protected-access data = csv._lookup_team_membership_data(self.course) self.assertEqual(len(data), 5) self.assert_teamset_membership(data[0], 'user1', 'audit', 'team_1_1', 'team_2_2', 'team_3_1') self.assert_teamset_membership(data[1], 'user2', 'verified', 'team_1_1', 'team_2_2', 'team_3_1') self.assert_teamset_membership(data[2], 'user3', 'honors', None, 'team_2_1', 'team_3_1') self.assert_teamset_membership(data[3], 'user4', 'masters', None, None, 'team_3_2') self.assert_teamset_membership(data[4], 'user5', 'masters', None, None, None)
def test_lookup_team_membership_data(self): with self.assertNumQueries(3): # pylint: disable=protected-access data = csv._lookup_team_membership_data(self.course) self._assert_test_users_on_team(_user_keyed_dict(data))