faults.py

from contextlib import contextmanager

@contextmanager
def injected_fault(params):
    fault_manager.inject(params)
    try:
        yield
    finally:
        fault_manager.clear(params)

def some_test():
    with injected_fault("disk_failure", offset):
        expect_exception(EIO, read file ...)