Function get_fodder_source

bpython/test/test_preprocess.py:18–33

(test_name)

def get_fodder_source(test_name):
    pattern = rf"#StartTest-{test_name}\n(.*?)#EndTest"
    orig, xformed = (
        re.search(pattern, inspect.getsource(module), re.DOTALL)
        for module in [original, processed]
    )

    if not orig:
        raise ValueError(
            f"Can't locate test {test_name} in original fodder file"
        )
    if not xformed:
        raise ValueError(
            f"Can't locate test {test_name} in processed fodder file"
        )
    return orig.group(1), xformed.group(1)


class TestPreprocessing(unittest.TestCase):