def test_Table_get_next_player(self):
"""Test Table.get_next_player() method."""
players = [ Player(name="One", stack=100),
Player(name="Two", stack=200),
Player(name="Three", stack=500),
Player(name="Four", stack=0)
]
table = Table()
table.seat_players(players, in_order=True)
self.assertEqual(table.get_next_player(players[0]), players[1])
self.assertEqual(table.get_next_player(players[1]), players[2])
self.assertEqual(table.get_next_player(players[2]), players[3])
self.assertEqual(table.get_next_player(players[3]), players[0])
self.assertEqual(
table.get_next_player(players[0], filter=lambda p: p.stack > 300),
players[2])
table.set_dealer(players[1])
self.assertEqual(table.get_dealer(), players[1])
self.assertEqual(str(table), "1: One 2: Two* 3: Three 4: Four")
