def test_generate_output_file(self):
self.assertEquals('2 5 6', main.generate_output_file(main.find_group_of_pupils('input.txt')))
def test_find_group_of_pupils(self):
self.assertEquals([2, 5, 6], main.find_group_of_pupils('input.txt'))
