TestCases class

test_cases.py

class TestCases:

    def __init__(self, **defaults):
        self._defaults = defaults
        self._cases = []
        self._next_index = 0

    def __iter__(self):
        return self

    def __next__(self):
        try:
            case = self._cases[self._next_index]
        except IndexError:
            raise StopIteration
        self._next_index += 1
        return Namespace(_customizations=case, **ChainMap(case, self._defaults))

    def add_case(self, **case):
        self._cases.append(case)

    def add_cases(self, *cases):
        for case in cases:
            self.add_case(**case)