def test_use_draw_cards(self): draw_card1 = Cards.WellsFargoCard(Cards.Suit.CLUBS, 5) draw_card2 = Cards.StageCoachCard(Cards.Suit.HEARTS, 8) player_names = [ 'Samuel', 'Gustav', 'Maria', 'Andrea', 'Tomas', 'Mona', 'Lennart' ] game_object = Game(player_names) game_object.active_player.add_cards([draw_card1, draw_card2]) self.assertEqual(len(game_object.active_player.hand), 2) game_object.use_card(draw_card1) self.assertEqual(len(game_object.active_player.hand), 4) game_object.use_card(draw_card2) self.assertEqual(len(game_object.active_player.hand), 5)
def create_deck(self): """ Create all the different cards an place in deck :return: """ deck = [] for s in Suit: for i in range(2, 9): deck.append(Cards.PangCard(i, s)) deck.append(Cards.MissCard(i, s)) for i in range(2, 9, 3): deck.append(Cards.PistolCard(i, s)) deck.append(Cards.StageCoachCard(i, s)) deck.append(Cards.BeerCard(i, s)) deck.append(Cards.PanicCard(i, s)) for i in range(3, 9, 2): deck.append(Cards.RifleCard(i, s)) deck.append(Cards.ScopeCard(i, s)) deck.append(Cards.BarrelCard(i, s)) deck.append(Cards.HorseCard(i, s)) for i in range(2, 9, 3): deck.append(Cards.WellsFargoCard(i, s)) return deck