¿Puedo forzar la depuración de python en AssertionError?

Question

Supongamos que tengo un programa python donde assert se ha usado para definir cómo deberían ser las cosas, y me gustaría capturar anomalías con read-eval-loop en lugar de tener AssertionError.

Por supuesto, podría haber

if (reality!=expectation): 
    print("assertion failed"); 
    import pdb; pdb.set_trace(); 

pero eso es mucho más feo en el código de una llanura assert(reality==expectation).

Podría tener pdb.set_trace() llamado en un bloque except: en el nivel superior, pero entonces habría perdido todo el contexto de la falla, ¿no? (Quiero decir, stacktrace podría recuperarse del objeto de excepción, pero no de los valores de argumento, etc.)

¿Hay algo como un indicador de línea de comandos --magic que podría convertir el intérprete python3 en lo que necesito?

Answer 1

Eche un vistazo al proyecto nose. Puede usarlo con el --pdb option para colocarlo en el depurador en caso de errores.

Answer 2

Principalmente tomado de this great snippet:

import sys 

def info(type, value, tb): 
    if hasattr(sys, 'ps1') or not sys.stderr.isatty() or type != AssertionError: 
     # we are in interactive mode or we don't have a tty-like 
     # device, so we call the default hook 
     sys.__excepthook__(type, value, tb) 
    else: 
     import traceback, pdb 
     # we are NOT in interactive mode, print the exception... 
     traceback.print_exception(type, value, tb) 
     print 
     # ...then start the debugger in post-mortem mode. 
     pdb.pm() 

sys.excepthook = info 

Al inicializar el código con esto, todos AssertionError s debe invocar AP.