Hola, yo hice lo siguiente: definí una variable global TEST con False en general. Pero, cuando hice los tests la cambié a True e hice los checkeos con
if Test:
No se si me explico, pero propongo esto como una solución viable sin requerir tanto trabajo.