1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83

import unittest
from types import FunctionType

class Paramaterizer(type):
    def __new__(meta, class_name, bases, attrs):
        
        for name, item in attrs.items():
            if not isinstance(item, FunctionType):
                continue
            
            params = getattr(item, 'params', None)
            if params is None:
                continue
            
            for index, args in enumerate(params):
                def test(self, args=args, name=name):
                    assertMethod = getattr(self, name)
                    assertMethod(**args)
                test.__doc__ = """%s with args: %s""" % (name, args)
                test_name = 'test_%s_%s' % (name, index + 1)
                test.__name__ = test_name
                
                if test_name in attrs:
                    raise Exception('Test class %s already has a method called: %s' % 
                                    (class_name, test_name))
                attrs[test_name] = test
        
        return type.__new__(meta, class_name, bases, attrs)
        
def with_params(params):
    def decorate(func):
        func.params = params
        return func
    return decorate

class TestCaseWithParams(unittest.TestCase):
    __metaclass__ = Paramaterizer
    
class Test(TestCaseWithParams):
    
    @with_params([dict(a=1, b=2), dict(a=3, b=3), dict(a=5, b=4)])
    def assertEqualWithParams(self, a, b):
        self.assertEqual(a, b)

    @with_params([dict(a=1, b=0), dict(a=3, b=2)])
    def assertZeroDivisionWithParams(self, a, b):
        self.assertRaises(ZeroDivisionError, lambda: a/b)
        
        

def test_generator(func):
    def inner(self):
        failures = []
        errors = []
        for test, args in func(self):
            try:
                test(*args)
            except self.failureException, e:
                failures.append((test.__name__, args, e))
            except Exception, e:
                errors.append((test.__name__, args, e))
        msg = '\n'.join('%s%s: %s: %s' % (name, args, e.__class__.__name__, e) for (name, args, e) in failures + errors) 
        if errors:
            raise Exception(msg)
        raise self.failureException(msg)
    return inner


class Test2(unittest.TestCase):

    @test_generator
    def testSomething(self):
        for a, b in ((1, 2), (3, 3), (5, 4)):
            yield self.assertEqual, (a, b)
        
        def raises():
            raise Exception('phooey')
        yield raises, ()
    

if __name__ == '__main__':
    unittest.main(testRunner=unittest.TextTestRunner(verbosity=2))