def testDiseaseLimit(self): g = game.build_game() g.cities = {name: game.City(0, 0, 0, 0) for name in game.infection_cards.keys()} for c in g.cities.keys()[:24]: game.infect_city(g, c, game.RED, 1, []) self.assertRaises(game.DefeatException, game.infect_city, g, g.cities.keys()[24], game.RED, 1, [])
def testOutbreak(self): g = game.build_game() g.cities = {name: game.City(0, 0, 0, 0) for name in game.infection_cards.keys()} game.infect_city(g, 'london', game.BLACK, 3, []) game.infect_city(g, 'london', game.BLACK, 1, []) self.assertEquals(1, g.outbreaks) self.assertEquals(3, g.cities['london'].black) self.assertEquals(1, g.cities['essen'].black) self.assertEquals(1, g.cities['paris'].black) self.assertEquals(1, g.cities['new york'].black) self.assertEquals(1, g.cities['madrid'].black) game.infect_city(g, 'madrid', game.BLACK, 2, []) self.assertEquals(3, g.cities['madrid'].black) game.infect_city(g, 'madrid', game.BLACK, 1, []) self.assertEquals(3, g.outbreaks) self.assertEquals(3, g.cities['madrid'].black) self.assertEquals(3, g.cities['london'].black) self.assertEquals(3, g.cities['new york'].black) self.assertRaises(game.DefeatException, game.infect_city, g, 'new york', game.BLACK, 1, [])