Scratch.py

In [7]:
def mytest():
    print "running test"
    def cleanup():
        print "cleaning up test"
    raise Exception({'msg': 'oh noes', 'cleanup': cleanup})

def pytest_exception_interact(e):
    print "handling error"
    e.message['cleanup']()


def pytest_test_runner(test):
    try:
        print "starting test"
        test()
    except Exception as e:
        print "test failed"
        pytest_exception_interact(e)
    finally:
        print "pytest cleaning up"

pytest_test_runner(mytest)
starting test
running test
test failed
handling error
cleaning up test
pytest cleaning up