def test_Pot_distribute(self):
"""Test Pot.distribute() method"""
player_one = Player(name="One")
player_two = Player(name="Two")
player_three = Player(name="Three")
player_four = Player(name="Four")
contending_players = [player_one, player_two, player_three, player_four]
pot = Pot(contending_players)
pot.amount = 200
pot.distribute(high_winners = [player_two])
self.assertEqual(pot.amount, 0)
self.assertEqual(player_two.stack, 200)
for player in contending_players:
player.stack = 0
pot.amount = 200
pot.distribute(high_winners = [player_one], low_winners = [player_four])
self.assertEqual(pot.amount, 0)
self.assertEqual(player_one.stack, 100)
self.assertEqual(player_four.stack, 100)
for player in contending_players:
player.stack = 0
pot.amount = 200
pot.distribute(high_winners = [player_one, player_two],
low_winners = [player_three])
self.assertEqual(pot.amount, 0)
self.assertEqual(player_one.stack, 50)
self.assertEqual(player_two.stack, 50)
self.assertEqual(player_three.stack, 100)
