def test_adv_input(self): line = 'Vampiries: 0, Zombies: 1, Witches: 12, Houses: 0' self.assertEqual( ({'Vampiries': 0, 'Zombies': 1, 'Witches': 12}, 0), main.parse_input(line) )
def test_simple_input(self): line = 'Vampiries: 1, Zombies: 1, Witches: 1, Houses: 1' self.assertEqual( ({'Vampiries': 1, 'Zombies': 1, 'Witches': 1},1), main.parse_input(line) )
def test_example2(self): line = 'Vampires: 3, Zombies: 2, Witches: 1, Houses: 10' costumes, houses = main.parse_input(line) self.assertEquals(36, main.solve(costumes, houses))