本文介绍如何在Python的unittest框架中实现参数化测试。通过定义基类并在子类中设置特定属性来实现不同测试用例的参数化。这种方法避免了重复代码，并使测试更加灵活。
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()
扫一扫
839
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
