def test_zero_vampires(self): self.assertEquals(0, main.solve({main.VAMPIRES:0}, 1))
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))
def test_no_costumes_houses(self): self.assertEquals(0, main.solve({}, 1))
def test_single_vampire_in_one_house(self): self.assertEquals( main.PRIZES[main.VAMPIRES], main.solve({main.VAMPIRES: 1}, 1) )
def test_zero_houses(self): self.assertEquals(0, main.solve({}, 0))