def remove_output_folder():
    """Remove the output folder after test."""
    yield
    if os.path.exists('output_folder'):
        utils.rmtree('output_folder')
Remove the output folder after test.
tests/test_output_folder.py