def test_add(capsys):
main = Main()
main.add_player("Carl")
out, err = capsys.readouterr()
assert ((Player("Carl", "b") in main.players)
or (Player("Carl", "y") in main.players))
assert ("Added: Player(name='Carl', color='y')\n" == out
or "Added: Player(name='Carl', color='b')\n" == out)
def test_no_add(capsys):
main = Main()
main.players = {
Player("Alice", "g"),
Player("Bob", "r"),
Player("Carl", "y"),
Player("Donna", "b")
}
main.add_player("Eric")
out, err = capsys.readouterr()
assert "Can't have more than 4 players\n" == out
