Jokenpo

main.py

def jokenpo(jogada):
    wins = {
        "pedra": "papel",
        "papel": "tesoura",
        "tesoura": "pedra"
    }
    if jogada not in wins:
        raise AssertionError
    return wins[jogada]

  
# código antes de ser refatorado:
def jokenpo(jogada):
    if jogada == "pedra":
        return "papel"
    elif jogada == "papel":
        return "tesoura"
    elif jogada == "tesoura":
        return "pedra"

test.py

from unittest import TestCase
import unittest
from main import jokenpo
"""
baseado na jogada do jokenpo, retorne a jogada vencedora
"""

class Test(TestCase):
    def test_pedra(self):
        self.assertEquals(jokenpo("pedra"), "papel")

    def test_papel(self):
        self.assertEquals(jokenpo("papel"), "tesoura")

    def test_tesoura(self):
        self.assertEquals(jokenpo("tesoura"), "pedra")

    def test_fail(self):
        with self.assertRaises(AssertionError):
            jokenpo("aaaaaa")

if __name__ == "__main__":
    unittest.main()