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))