Tirsvad
d0e6c93573
TirSystem/github-action: Sync GitHub mirror metadata / sync-metadata (push) Successful in 7s
54 lines
1.6 KiB
Python
54 lines
1.6 KiB
Python
import pytest
|
|
|
|
from rock_paper_scissors import game
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
("user", "computer", "expected"),
|
|
[
|
|
(0, 0, game.DRAW),
|
|
(1, 1, game.DRAW),
|
|
(2, 2, game.DRAW),
|
|
(0, 2, game.WIN), # rock beats scissors
|
|
(1, 0, game.WIN), # paper beats rock
|
|
(2, 1, game.WIN), # scissors beats paper
|
|
(2, 0, game.LOSE),
|
|
(0, 1, game.LOSE),
|
|
(1, 2, game.LOSE),
|
|
],
|
|
)
|
|
def test_determine_result(user, computer, expected):
|
|
assert game.determine_result(user, computer) == expected
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
("text", "expected"),
|
|
[("0", 0), (" 1 ", 1), ("2", 2), ("3", None), ("-1", None), ("rock", None), ("", None)],
|
|
)
|
|
def test_parse_choice(text, expected):
|
|
assert game.parse_choice(text) == expected
|
|
|
|
|
|
def test_main_invalid_input(monkeypatch, capsys):
|
|
monkeypatch.setattr("builtins.input", lambda _: "9")
|
|
game.main()
|
|
assert "invalid" in capsys.readouterr().out
|
|
|
|
|
|
def test_main_win(monkeypatch, capsys):
|
|
monkeypatch.setattr("builtins.input", lambda _: "0")
|
|
monkeypatch.setattr(game.random, "randint", lambda a, b: 2)
|
|
game.main()
|
|
assert "You win!" in capsys.readouterr().out
|
|
|
|
|
|
def test_colorize_title_disabled_returns_plain():
|
|
assert game.colorize_title(enabled=False) == game.TITLE
|
|
|
|
|
|
def test_colorize_title_enabled_colors_each_word():
|
|
out = game.colorize_title(enabled=True)
|
|
for color in game.TITLE_COLORS:
|
|
assert color in out
|
|
assert out.count(game.RESET) == len(game.TITLE.strip().splitlines())
|