def test_by_size(self): [ factories.ScheduleABySizeFactory( committee_id=self.committees[0].committee_id, cycle=2012, total=50, size=200, ), factories.ScheduleABySizeFactory( committee_id=self.committees[1].committee_id, cycle=2012, total=150, size=200, ), ] results = self._results( api.url_for( ScheduleABySizeCandidateView, candidate_id=self.candidate.candidate_id, cycle=2012, )) self.assertEqual(len(results), 1) expected = { 'candidate_id': self.candidate.candidate_id, 'cycle': 2012, 'total': 200, 'size': 200, } self.assertEqual(results[0], expected)
def test_by_size(self): [ factories.ScheduleABySizeFactory( committee_id=self.committees[0].committee_id, cycle=2012, total=50, size=200, count=20, ), factories.ScheduleABySizeFactory( committee_id=self.committees[1].committee_id, cycle=2012, total=150, size=200, count=20, ), ] results = self._results( api.url_for( ScheduleABySizeCandidateView, candidate_id=self.candidate.candidate_id, cycle=2012, )) assert len(results) == 1 expected = { 'candidate_id': self.candidate.candidate_id, 'cycle': 2012, 'total': 200, 'size': 200, 'count': 20, } assert results[0] == expected