Python,单元测试 - 将命令行参数传递给unittest.TestCase的setUp

import sys
import unittest

class TestStringMethods(unittest.TestCase):

    def setUp(self):
        # 获取命令行参数
        args = sys.argv[1:]
        # 解析参数
        self.arg1 = None
        self.arg2 = None
        for arg in args:
            if arg.startswith('--arg1='):
                self.arg1 = arg.split('=')[1]
            elif arg.startswith('--arg2='):
                self.arg2 = arg.split('=')[1]

    def test_upper(self):
        self.assertEqual('foo'.upper(), 'FOO')
        self.assertEqual('bar'.upper(), 'BAR')

    def test_isupper(self):
        self.assertTrue('FOO'.isupper())
        self.assertFalse('Bar'.isupper())

    def test_split(self):
        s = 'hello world'
        self.assertEqual(s.split(), ['hello', 'world'])
        # 使用arg1和arg2参数
        self.assertEqual(s.split(self.arg1), ['hello', 'world'])

if __name__ == '__main__':
    unittest.main()

python -m unittest test_module.TestStringMethods.test_upper --arg1=value1 --arg2=value2