def test_correct_win(self): door_1, door_2, door_3 = Door(False), Door(True), Door(False) game = Game(door_1, door_2, door_3) game.contestant_guess = door_2 self.assertEqual(True, game.contestant_has_won())