在python中，我如何将命令行中的参数传递给unittest函数。这是到目前为止的代码……我知道这是错误的。
class TestingClass(unittest.TestCase):
def testEmails(self):
assertEqual(email_from_argument, "my_email@example.com")
if __name__ == "__main__":
unittest.main(argv=[sys.argv[1]])
email_from_argument = sys.argv[1]
Answers:
因此，这里的医生说：“你说那很疼？那就不要那样做！” 可能是对的。但是，如果您确实愿意，这是将参数传递给单元测试的一种方法：
import sys
import unittest
class MyTest(unittest.TestCase):
USERNAME = "jemima"
PASSWORD = "password"
def test_logins_or_something(self):
print('username : {}'.format(self.USERNAME))
print('password : {}'.format(self.PASSWORD))
if __name__ == "__main__":
if len(sys.argv) > 1:
MyTest.USERNAME = sys.argv.pop()
MyTest.PASSWORD = sys.argv.pop()
unittest.main()
这将使您可以：
python mytests.py ausername apassword
您需要argv.pops，以便您的命令行参数不会与unittest自己的混乱。
[更新]您可能要研究的另一件事是使用环境变量：
import os
import unittest
class MyTest(unittest.TestCase):
USERNAME = "jemima"
PASSWORD = "password"
def test_logins_or_something(self):
print('username : {}'.format(self.USERNAME))
print('password : {}'.format(self.PASSWORD))
if __name__ == "__main__":
MyTest.USERNAME = os.environ.get('TEST_USERNAME', MyTest.USERNAME)
MyTest.PASSWORD = os.environ.get('TEST_PASSWORD', MyTest.PASSWORD)
unittest.main()
这样您就可以运行：
TEST_USERNAME=ausername TEST_PASSWORD=apassword python mytests.py
而且它的优点是您不会搞砸unittest自己的参数解析。缺点是它在Windows上无法像...
sys.argv.pop()方法似乎不适用于python3。我很想知道如何在中完成此操作python3。
尽管您不应该做出正确的评论，但对于那些确实想要这样做的人来说，另一种方法是：
import unittest
class MyTest(unittest.TestCase):
def __init__(self, testName, extraArg):
super(MyTest, self).__init__(testName) # calling the super class init varies for different python versions. This works for 2.7
self.myExtraArg = extraArg
def test_something(self):
print(self.myExtraArg)
# call your test
suite = unittest.TestSuite()
suite.addTest(MyTest('test_something', extraArg))
unittest.TextTestRunner(verbosity=2).run(suite)
即使测试专家说我们不应该这样做：我愿意。在某些情况下，有意义的是具有参数来朝正确的方向驱动测试，例如：
对我而言，使用环境变量足以满足此需求，因为您不必编写专用代码来传递参数。它受Python支持。干净简单。
当然，我不主张进行完全可参数化的测试。但是我们必须务实，正如我所说，在某些情况下，您需要一个或两个参数。我们不应该反对它:)
import os
import unittest
class MyTest(unittest.TestCase):
def setUp(self):
self.var1 = os.environ["VAR1"]
self.var2 = os.environ["VAR2"]
def test_01(self):
print("var1: {}, var2: {}".format(self.var1, self.var2))
然后从命令行（在Linux上测试）
$ export VAR1=1
$ export VAR2=2
$ python -m unittest MyTest
var1: 1, var2: 2
.
----------------------------------------------------------------------
Ran 1 test in 0.000s
OK
如果您想将steffens21的方法用于unittest.TestLoader，则可以修改原始的测试加载器（请参阅参考资料unittest.py）：
import unittest
from unittest import suite
class TestLoaderWithKwargs(unittest.TestLoader):
"""A test loader which allows to parse keyword arguments to the
test case class."""
def loadTestsFromTestCase(self, testCaseClass, **kwargs):
"""Return a suite of all tests cases contained in
testCaseClass."""
if issubclass(testCaseClass, suite.TestSuite):
raise TypeError("Test cases should not be derived from "\
"TestSuite. Maybe you meant to derive from"\
" TestCase?")
testCaseNames = self.getTestCaseNames(testCaseClass)
if not testCaseNames and hasattr(testCaseClass, 'runTest'):
testCaseNames = ['runTest']
# Modification here: parse keyword arguments to testCaseClass.
test_cases = []
for test_case_name in testCaseNames:
test_cases.append(testCaseClass(test_case_name, **kwargs))
loaded_suite = self.suiteClass(test_cases)
return loaded_suite
# call your test
loader = TestLoaderWithKwargs()
suite = loader.loadTestsFromTestCase(MyTest, extraArg=extraArg)
unittest.TextTestRunner(verbosity=2).run(suite)