Decorator for unit test methods allowing parametrized execution of tests. Inspiring by PHPUnit's @dataProvider.

data_provider.py

import itertools

class data_set:
    """
    Class: data_set
    
    Allows to combine args and kwargs in set of parameters.
    """

    def __init__(self, *args, **kwargs):
        self.args = args
        self.kwargs = kwargs


def data_provider(*data_sets, **kw_data_sets):
    """
    Decorator: data_provider

    Allows to write parametrized tests.

    Usage:
    @data_provider(
        (1, 2, 3),
        {'a': 4, 'b': 5, 'c': 6},
        ds789=data_set(7, 8, c=9)
        # more parameter sets...
    )
    def some_test(self, a, b, c):
        # testing code...
    """
    def _decorator(test_method):

        def _executor(test_case):
            for ds in itertools.chain(data_sets, kw_data_sets.itervalues()):
                test_case.setUp()
                try:
                    if isinstance(ds, dict):
                        test_method(test_case, **ds)
                    elif isinstance(ds, data_set):
                        test_method(test_case, *ds.args, **ds.kwargs)
                    else:
                        test_method(test_case, *ds)
                finally:
                    test_case.tearDown()
                    
        return _executor
        
    return _decorator