# 
from common import Params
from evolution import Universe
from particles import Particle
from library.SM import (particles as SMP, interactions as SMI)


eps = 1e-5
# 
def setup():
    params = Params(T=SMP.leptons.neutrino_e['decoupling_temperature'],
                    dy=0.025)
    return [params], {}
# 
def non_equilibium_setup():
    args, _ = setup()
    params = args[0]

    photon = Particle(**SMP.photon)
    neutrino_e = Particle(**SMP.leptons.neutrino_e)
    neutrino_mu = Particle(**SMP.leptons.neutrino_mu)
    neutrino_self_scattering = SMI.neutrino_scattering(neutrino_e, neutrino_e)

    universe = Universe(params=params, plotting=False)
    universe.add_particles([photon, neutrino_e, neutrino_mu])
    universe.interactions += [neutrino_self_scattering]

    return [params, universe], {}
#

Decorator to add setup and/or teardown methods to a test function::

@with_setup_args(setup, teardown) def test_something(): " ... "

The setup function should return (args, kwargs) which will be passed to test function, and teardown function.

Note that with_setup_args is useful only for test functions, not for test methods or inside of TestCase subclasses.

def with_setup_args(setup, teardown=None):

    def decorate(func):
        args = []
        kwargs = {}
# 
        def test_wrapped():
            func(*args, **kwargs)

        test_wrapped.__name__ = func.__name__
# 
        def setup_wrapped():
            a, k = setup()
            args.extend(a)
            kwargs.update(k)
            if hasattr(func, 'setup'):
                func.setup()
        test_wrapped.setup = setup_wrapped

        if teardown:

            def teardown_wrapped():
                if hasattr(func, 'teardown'):
                    func.teardown()
                teardown(*args, **kwargs)

            test_wrapped.teardown = teardown_wrapped
# 
        else:
            if hasattr(func, 'teardown'):
                test_wrapped.teardown = func.teardown()
        return test_wrapped
# 
    return decorate