Пример #1
0
    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)
        )
Пример #2
0
    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)
        )
Пример #3
0
    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))