def test_generator_proportion_0_gives_no_sos(self): m = Member() m.sos_percentage = 0 generator = Generator([m], self._basic_mock_work_day_service, 10) self.assertEqual(len(generator.pot), 0)
def test_generator_proportion_50_gives_one_sos(self): m = Member() m.sos_percentage = 50 generator = Generator([m], self._basic_mock_work_day_service, 10) generator._populate_pot() self.assertEqual(len(generator.pot), 1)