tests/testutils/patch.py

import subprocess


def apply(file, patch):
    try:
        subprocess.check_output(['patch', file, patch])
    except subprocess.CalledProcessError as e:
        message = "Patch failed with exit code {}\n Output:\n {}".format(
            e.returncode, e.output)
        print(message)
        raise


def remove(file, patch):
    try:
        subprocess.check_output(['patch', '--reverse', file, patch])
    except subprocess.CalledProcessError as e:
        message = "patch --reverse failed with exit code {}\n Output:\n {}".format(
            e.returncode, e.output)
        print(message)
        raise