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