def ubuntu20(function):
    """Decorator to set the Linux distribution to Ubuntu 20.04"""
    def wrapper(*args, **kwargs):
        hpccm.config.g_linux_distro = linux_distro.UBUNTU
        hpccm.config.g_linux_version = Version('20.04')
        return function(*args, **kwargs)

    return wrapper
Decorator to set the Linux distribution to Ubuntu 20.04
test/helpers.py