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 testInit(self): g = game.build_game() self.assertEquals([game.ATLANTA], g.research_stations) self.assertEquals(4, len(g.players)) self.assertEquals(5, len([card for card in g.player_draw_pile if card == game.EPIDEMIC])) for player in g.players: self.assertEquals(2, len(player.cards)) self.assertEquals(game.ATLANTA, player.location) self.assertEquals(48, len(g.cities)) self.assertEquals(0, len(g.player_discard_pile)) self.assertEquals(9, len(g.infection_discard_pile))
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, [])