def test_Pot(self):
"""Test Pot class"""
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)
self.assertIsNotNone(pot)
self.assertEqual(pot.amount, 0) # Default amount
self.assertIsNone(pot.parent)
self.assertListEqual(pot.contending_players, contending_players)
pot.fold_player(player_one)
contending_players.remove(player_one)
self.assertListEqual(pot.contending_players, contending_players)
self.assertIn(player_one, pot.folded_players)
pot.amount += 100
s = str(pot)
self.assertIsInstance(s, str)
self.assertEqual(s, "Main pot 100 (contenders: Two,Three,Four)")
contending_players.remove(player_three)
pot.new_side_pot(contending_players)
self.assertListEqual(pot.contending_players, contending_players)
self.assertEqual(pot.amount, 0)
self.assertIsNotNone(pot.parent)
self.assertEqual(pot.parent.amount, 100)
# Should remove from both pot and pot.parent
pot.fold_player(player_four)
self.assertListEqual(pot.contending_players, [player_two])
self.assertNotIn(player_four, pot.parent.contending_players)
self.assertIn(player_four, pot.parent.folded_players)
s = str(pot)
self.assertIsInstance(s, str)
self.assertEqual(s, "Side pot 0 (contenders: Two)")
def test_Pot_new_side_pot(self):
"""Test Pot.new_side_pot() 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)
player_one.bet = 100
player_two.bet = 0
player_three.bet = 120
player_four.bet = 90
pot.new_side_pot()
# Should have dropped player_two since .bet == 0
self.assertListEqual(pot.contending_players,
[player_one, player_three, player_four])
