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