python testcase,实例的Python unittest.TestCase的带参数

I would like to iterate over a list of items, and run an assertion on each of them. One example might be checking whether each number in a list is odd.

TestCase:

class TestOdd(unittest.TestCase):

def runTest(self):

"""Assert that the item is odd"""

self.assertTrue( NUMBER %2==1, "Number should be odd")

Test suite:

if __name__ == '__main__':

suite = unittest.TestSuite()

suite.addTest(TestOdd())

# I would like to have:

# suite.addTest(TestOdd(1))

# suite.addTest(TestOdd(2))

# suite.addTest(TestOdd(3))

# ...

unittest.main()

How can I instantiate a TestOdd object with an argument - for example, the number to be tested?

Update: According to a blog post from 2011 (posted as answer), there is no built-in mechanism for parametrized tests. I will be happy to accept any cleaner solutions.

解决方案

Same can be achieved using class attributes.

class TestOdd1(unittest.TestCase):

NUMBER=1

def runTest(self):

"""Assert that the item is odd"""

self.assertTrue( self.NUMBER % 2 == 1, "Number should be odd")

class TestOdd2(TestOdd1):

NUMBER=2

if __name__ == '__main__':

unittest.main()

The unittesting will discover them automatically, so no need to create a suite.

If you want to avoid using a TestCase for base class, you can use multiple inheritance:

from unittest import TestCase, main

class TestOdd:

def runTest(self):

"""Assert that the item is odd"""

self.assertTrue( self.NUMBER % 2 == 1, "Number should be odd")

class TestOdd1(TestOdd, TestCase):

NUMBER=1

class TestOdd2(TestOdd, TestCase):

NUMBER=2

if __name__ == '__main__':

main()