def test_incorrect(self):
"""
Test for situation when user loses
"""
hangman = Hangman("hello")
self.assertFalse(hangman.handle_letter('a'))
self.assertFalse(hangman.handle_letter('a'))
self.assertFalse(hangman.handle_letter('a'))
self.assertFalse(hangman.handle_letter('a'))
self.assertFalse(hangman.handle_letter('a'))
self.assertTrue(hangman.check_lose())
def test_correct(self):
"""
Test for situation when user wins
"""
hangman = Hangman("hello")
self.assertTrue(hangman.handle_letter('l'))
self.assertFalse(hangman.handle_letter('a'))
self.assertTrue(hangman.handle_letter('o'))
self.assertTrue(hangman.handle_letter('h'))
self.assertTrue(hangman.handle_letter('e'))
self.assertTrue(hangman.check_win())
self.assertEqual(hangman.get_num_mistakes(), 1)
self.assertEqual(hangman.get_current_word(), "hello")
