def run_with_api(self, code): """Run code with didyoumean after enabling didyoumean hook.""" prev_hook = sys.excepthook self.assertEqual(prev_hook, sys.excepthook) didyoumean_enablehook() self.assertNotEqual(prev_hook, sys.excepthook) try: no_exception(code) except: last_type, last_value, last_traceback = sys.exc_info() sys.excepthook(last_type, last_value, last_traceback) raise finally: self.assertNotEqual(prev_hook, sys.excepthook) didyoumean_disablehook() self.assertEqual(prev_hook, sys.excepthook)
def run_with_api(self, code): """Run code with didyoumean after enabling didyoumean hook.""" prev_handler = None shell = DummyShell() module = sys.modules['didyoumean_api'] shell.set(module) self.assertEqual(shell.handler, prev_handler) didyoumean_enablehook() self.assertNotEqual(shell.handler, prev_handler) try: no_exception(code) except: shell.showtraceback(sys.exc_info()) raise finally: self.assertNotEqual(shell.handler, prev_handler) didyoumean_disablehook() self.assertEqual(shell.handler, prev_handler) shell.remove(module) shell = None