def test_correct_answer_fourth_try_wins(self):
solution = [Peg("Green"), Peg("Blue"), Peg("Pink"), Peg("Yellow")]
game = Game(solution)
incorrect_guess = [Peg("Green"), Peg("Green"), Peg("Green"), Peg("Green")]
for i in range(0, 3):
result = game.make_guess(incorrect_guess)
result, _ = game.make_guess(solution)
self.assertEqual("Win!!!", result)
def test_guess_with_correct_color_and_position_returns_black_peg(self):
# Black key peg is placed for each code peg from the guess which is correct in both color and position
solution = [Peg("Blue"), Peg("Blue"), Peg("Pink"), Peg("Pink")]
game = Game(solution)
result, hint = game.make_guess([Peg("Green"), Peg("Green"), Peg("Green"), Peg("Pink")])
self.assertEqual(hint, [None, None, None, Peg("Black")],
msg=f'Answer with correct color and position should return black peg')
def test_guess_with_duplicate_colors_gives_hint(self):
solution = [Peg("Blue"), Peg("Blue"), Peg("Pink"), Peg("Blue")]
game = Game(solution)
result, hint = game.make_guess([Peg("Blue"), Peg("Blue"), Peg("Blue"), Peg("Yellow")])
self.assertEqual(hint, [Peg("Black"), Peg("Black"), Peg("White"), None])
def test_no_hint_when_guess_colors_exceed_solution(self):
solution = [Peg("Blue"), Peg("Blue"), Peg("Pink"), Peg("Pink")]
game = Game(solution)
result, hint = game.make_guess([Peg("Blue"), Peg("Blue"), Peg("Blue"), Peg("Pink")])
correct_hint = [Peg("Black"), Peg("Black"), None, Peg("Black")]
self.assertEqual(hint, correct_hint, msg=f"Answer with more than correct color matches should not return white peg")
def test_guess_with_correct_color_returns_white_peg(self):
solution = [Peg("Blue"), Peg("Blue"), Peg("Pink"), Peg("Pink")]
game = Game(solution)
result, hint = game.make_guess([Peg("Green"), Peg("Green"), Peg("Blue"), Peg("Green")])
self.assertEqual(hint, [None, None, Peg("White"), None],
msg=f'Answer with correct color should return white peg: {hint}')
def test_fifteen_incorrect_guesses_is_loss(self):
game = Game([Peg("Green"), Peg("Blue"), Peg("Pink"), Peg("Yellow")])
incorrect_guess = [Peg("Green"), Peg("Green"), Peg("Green"), Peg("Green")]
for i in range(0, 15):
result, _ = game.make_guess(incorrect_guess)
self.assertEqual("Loss, too many guesses", result)
def test_ten_incorrect_guesses_is_loss(self):
game_two = Game([Peg("Green"), Peg("Blue"), Peg("Pink"), Peg("Yellow")])
for i in range(0, 10):
result, _ = game_two.make_guess([Peg("Green"), Peg("Green"), Peg("Green"), Peg("Green")])
self.assertEqual("Loss, too many guesses", result)
def test_generate_random_answer(self):
game_two = Game() # game without solution
guess = [Peg("Green"), Peg("Green"), Peg("Blue"), Peg("Blue")]
result, _ = game_two.make_guess(guess)
self.assertNotEqual("Win!!!", result)
def test_incorrect_answer_does_not_win(self):
game_one = Game([Peg("Green"), Peg("Green"), Peg("Blue"), Peg("Blue")])
result, _ = game_one.make_guess([Peg("Green"), Peg("Green"), Peg("Green"), Peg("Green")])
self.assertNotEqual("Win!!!", result)