使用 unittest discover 传递参数（对于 argparse）
Passing arguments (for argparse) with unittest discover
foo是一个目录嵌套较深的Python项目，在各个子目录下包含~30个unittest个文件。在 foo 的 setup.py 中，我 added a custom "test" command 内部 运行ning
python -m unittest discover foo '*test.py'
请注意，这使用 unittest's discovery 模式。
由于一些测试非常慢，我最近决定测试应该 "levels"。 this question 的答案很好地解释了如何让 unittest 和 argparse 很好地协同工作。所以现在，我可以 运行 一个 个人 单元测试文件，比如 foo/bar/_bar_test.py，
python foo/bar/_bar_test.py --level=3
并且只有 3 级测试是 运行。
问题是我无法弄清楚如何传递自定义标志（在本例中使用发现“--level=3”。我尝试的一切都失败了，例如：
$ python -m unittest discover --level=3 foo '*test.py'
Usage: python -m unittest discover [options]
python -m unittest discover: error: no such option: --level
$ python -m --level=3 unittest discover foo '*test.py'
/usr/bin/python: No module named --level=3
如何将 --level=3 传递给各个单元测试？如果可能的话，我想避免将不同级别的测试划分到不同的文件中。
赏金编辑
赏金前（罚款）解决方案建议使用系统环境变量。这还不错，但我正在寻找更清洁的东西。
将多文件测试 运行ner（即 python -m unittest discover foo '*test.py'）更改为其他内容是可以的，只要：
- 它允许为多文件单元测试生成单个报告。
- 它可以以某种方式支持多个测试级别（使用问题中的技术，或使用其他一些不同的机制）。
使用 discover 时无法传递参数。
DiscoveringTestLoader class 从发现中删除所有不匹配的文件（使用'*test.py --level=3' 消除）并仅将文件名传递到 unittest.TextTestRunner
目前可能唯一的选择是使用环境变量
LEVEL=3 python -m unittest discoverfoo '*test.py'
你遇到的问题是 unittest 参数解析器根本不理解这种语法。因此，您必须在调用 unittest 之前删除参数。
一个简单的方法是创建一个包装器模块（比如 my_unittest.py）来查找您的额外参数，将它们从 sys.argv 中剥离，然后调用 unittest 中的主条目。
好消息是...该包装器的代码与您已经用于单个文件案例的代码基本相同！你只需要将它放入一个单独的文件中即可。
编辑：根据要求添加了下面的示例代码...
首先，新建文件到运行的UTs(my_unittest.py):
import sys
import unittest
from parser import wrapper
if __name__ == '__main__':
wrapper.parse_args()
unittest.main(module=None, argv=sys.argv)
现在 parser.py，它必须在一个单独的文件中，以避免在 __main__ 模块中，以便全局引用工作：
import sys
import argparse
import unittest
class UnitTestParser(object):
def __init__(self):
self.args = None
def parse_args(self):
# Parse optional extra arguments
parser = argparse.ArgumentParser()
parser.add_argument('--level', type=int, default=0)
ns, args = parser.parse_known_args()
self.args = vars(ns)
# Now set the sys.argv to the unittest_args (leaving sys.argv[0] alone)
sys.argv[1:] = args
wrapper = UnitTestParser()
最后是一个示例测试用例 (project_test.py)，用于测试参数是否被正确解析：
import unittest
from parser import wrapper
class TestMyProject(unittest.TestCase):
def test_len(self):
self.assertEqual(len(wrapper.args), 1)
def test_level3(self):
self.assertEqual(wrapper.args['level'], 3)
现在证明：
$ python -m my_unittest discover --level 3 . '*test.py'
..
----------------------------------------------------------------------
Ran 2 tests in 0.000s
OK
这不会使用 unittest discover 传递 args，但它会完成您正在尝试做的事情。
这是leveltest.py。将它放在模块搜索路径中的某处（可能是当前目录或站点包）：
import argparse
import sys
import unittest
# this part copied from unittest.__main__.py
if sys.argv[0].endswith("__main__.py"):
import os.path
# We change sys.argv[0] to make help message more useful
# use executable without path, unquoted
# (it's just a hint anyway)
# (if you have spaces in your executable you get what you deserve!)
executable = os.path.basename(sys.executable)
sys.argv[0] = executable + " -m leveltest"
del os
def _id(obj):
return obj
# decorator that assigns test levels to test cases (classes and methods)
def level(testlevel):
if unittest.level < testlevel:
return unittest.skip("test level too low.")
return _id
def parse_args():
parser = argparse.ArgumentParser()
parser.add_argument('--level', type=int, default=3)
ns, args = parser.parse_known_args(namespace=unittest)
return ns, sys.argv[:1] + args
if __name__ == "__main__":
ns, remaining_args = parse_args()
# this invokes unittest when leveltest invoked with -m flag like:
# python -m leveltest --level=2 discover --verbose
unittest.main(module=None, argv=remaining_args)
下面是如何在示例 testproject.py 文件中使用它：
import unittest
import leveltest
# This is needed before any uses of the @leveltest.level() decorator
# to parse the "--level" command argument and set the test level when
# this test file is run directly with -m
if __name__ == "__main__":
ns, remaining_args = leveltest.parse_args()
@leveltest.level(2)
class TestStringMethods(unittest.TestCase):
@leveltest.level(5)
def test_upper(self):
self.assertEqual('foo'.upper(), 'FOO')
@leveltest.level(3)
def test_isupper(self):
self.assertTrue('FOO'.isupper())
self.assertFalse('Foo'.isupper())
@leveltest.level(4)
def test_split(self):
s = 'hello world'
self.assertEqual(s.split(), ['hello', 'world'])
# check that s.split fails when the separator is not a string
with self.assertRaises(TypeError):
s.split(2)
if __name__ == '__main__':
# this invokes unittest when this file is executed with -m
unittest.main(argv=remaining_args)
然后您可以 运行 直接通过 运行ning testproject.py 进行测试，例如：
~roottwo\projects> python testproject.py --level 2 -v
test_isupper (__main__.TestStringMethods) ... skipped 'test level too low.'
test_split (__main__.TestStringMethods) ... skipped 'test level too low.'
test_upper (__main__.TestStringMethods) ... skipped 'test level too low.'
----------------------------------------------------------------------
Ran 3 tests in 0.000s
OK (skipped=3)
~roottwo\projects> python testproject.py --level 3 -v
test_isupper (__main__.TestStringMethods) ... ok
test_split (__main__.TestStringMethods) ... skipped 'test level too low.'
test_upper (__main__.TestStringMethods) ... skipped 'test level too low.'
----------------------------------------------------------------------
Ran 3 tests in 0.001s
OK (skipped=2)
~roottwo\projects> python testproject.py --level 4 -v
test_isupper (__main__.TestStringMethods) ... ok
test_split (__main__.TestStringMethods) ... ok
test_upper (__main__.TestStringMethods) ... skipped 'test level too low.'
----------------------------------------------------------------------
Ran 3 tests in 0.001s
OK (skipped=1)
~roottwo\projects> python testproject.py --level 5 -v
test_isupper (__main__.TestStringMethods) ... ok
test_split (__main__.TestStringMethods) ... ok
test_upper (__main__.TestStringMethods) ... ok
----------------------------------------------------------------------
Ran 3 tests in 0.001s
OK
通过像这样使用单元测试发现：
~roottwo\projects> python -m leveltest --level 2 -v
test_isupper (testproject.TestStringMethods) ... skipped 'test level too low.'
test_split (testproject.TestStringMethods) ... skipped 'test level too low.'
test_upper (testproject.TestStringMethods) ... skipped 'test level too low.'
----------------------------------------------------------------------
Ran 3 tests in 0.003s
OK (skipped=3)
~roottwo\projects> python -m leveltest --level 3 discover -v
test_isupper (testproject.TestStringMethods) ... ok
test_split (testproject.TestStringMethods) ... skipped 'test level too low.'
test_upper (testproject.TestStringMethods) ... skipped 'test level too low.'
----------------------------------------------------------------------
Ran 3 tests in 0.001s
OK (skipped=2)
~roottwo\projects> python -m leveltest --level 4 -v
test_isupper (testproject.TestStringMethods) ... ok
test_split (testproject.TestStringMethods) ... ok
test_upper (testproject.TestStringMethods) ... skipped 'test level too low.'
----------------------------------------------------------------------
Ran 3 tests in 0.001s
OK (skipped=1)
~roottwo\projects> python -m leveltest discover --level 5 -v
test_isupper (testproject.TestStringMethods) ... ok
test_split (testproject.TestStringMethods) ... ok
test_upper (testproject.TestStringMethods) ... ok
----------------------------------------------------------------------
Ran 3 tests in 0.001s
OK
或者将测试用例指定为运行，如：
~roottwo\projects>python -m leveltest --level 3 testproject -v
test_isupper (testproject.TestStringMethods) ... ok
test_split (testproject.TestStringMethods) ... skipped 'test level too low.'
test_upper (testproject.TestStringMethods) ... skipped 'test level too low.'
----------------------------------------------------------------------
Ran 3 tests in 0.002s
OK (skipped=2)
foo是一个目录嵌套较深的Python项目，在各个子目录下包含~30个unittest个文件。在 foo 的 setup.py 中，我 added a custom "test" command 内部 运行ning
python -m unittest discover foo '*test.py'
请注意，这使用 unittest's discovery 模式。
由于一些测试非常慢，我最近决定测试应该 "levels"。 this question 的答案很好地解释了如何让 unittest 和 argparse 很好地协同工作。所以现在，我可以 运行 一个 个人 单元测试文件，比如 foo/bar/_bar_test.py，
python foo/bar/_bar_test.py --level=3
并且只有 3 级测试是 运行。
问题是我无法弄清楚如何传递自定义标志（在本例中使用发现“--level=3”。我尝试的一切都失败了，例如：
$ python -m unittest discover --level=3 foo '*test.py'
Usage: python -m unittest discover [options]
python -m unittest discover: error: no such option: --level
$ python -m --level=3 unittest discover foo '*test.py'
/usr/bin/python: No module named --level=3
如何将 --level=3 传递给各个单元测试？如果可能的话，我想避免将不同级别的测试划分到不同的文件中。
赏金编辑
赏金前（罚款）解决方案建议使用系统环境变量。这还不错，但我正在寻找更清洁的东西。
将多文件测试 运行ner（即 python -m unittest discover foo '*test.py'）更改为其他内容是可以的，只要：
- 它允许为多文件单元测试生成单个报告。
- 它可以以某种方式支持多个测试级别（使用问题中的技术，或使用其他一些不同的机制）。
使用 discover 时无法传递参数。
DiscoveringTestLoader class 从发现中删除所有不匹配的文件（使用'*test.py --level=3' 消除）并仅将文件名传递到 unittest.TextTestRunner
目前可能唯一的选择是使用环境变量
LEVEL=3 python -m unittest discoverfoo '*test.py'
你遇到的问题是 unittest 参数解析器根本不理解这种语法。因此，您必须在调用 unittest 之前删除参数。
一个简单的方法是创建一个包装器模块（比如 my_unittest.py）来查找您的额外参数，将它们从 sys.argv 中剥离，然后调用 unittest 中的主条目。
好消息是...该包装器的代码与您已经用于单个文件案例的代码基本相同！你只需要将它放入一个单独的文件中即可。
编辑：根据要求添加了下面的示例代码...
首先，新建文件到运行的UTs(my_unittest.py):
import sys
import unittest
from parser import wrapper
if __name__ == '__main__':
wrapper.parse_args()
unittest.main(module=None, argv=sys.argv)
现在 parser.py，它必须在一个单独的文件中，以避免在 __main__ 模块中，以便全局引用工作：
import sys
import argparse
import unittest
class UnitTestParser(object):
def __init__(self):
self.args = None
def parse_args(self):
# Parse optional extra arguments
parser = argparse.ArgumentParser()
parser.add_argument('--level', type=int, default=0)
ns, args = parser.parse_known_args()
self.args = vars(ns)
# Now set the sys.argv to the unittest_args (leaving sys.argv[0] alone)
sys.argv[1:] = args
wrapper = UnitTestParser()
最后是一个示例测试用例 (project_test.py)，用于测试参数是否被正确解析：
import unittest
from parser import wrapper
class TestMyProject(unittest.TestCase):
def test_len(self):
self.assertEqual(len(wrapper.args), 1)
def test_level3(self):
self.assertEqual(wrapper.args['level'], 3)
现在证明：
$ python -m my_unittest discover --level 3 . '*test.py'
..
----------------------------------------------------------------------
Ran 2 tests in 0.000s
OK
这不会使用 unittest discover 传递 args，但它会完成您正在尝试做的事情。
这是leveltest.py。将它放在模块搜索路径中的某处（可能是当前目录或站点包）：
import argparse
import sys
import unittest
# this part copied from unittest.__main__.py
if sys.argv[0].endswith("__main__.py"):
import os.path
# We change sys.argv[0] to make help message more useful
# use executable without path, unquoted
# (it's just a hint anyway)
# (if you have spaces in your executable you get what you deserve!)
executable = os.path.basename(sys.executable)
sys.argv[0] = executable + " -m leveltest"
del os
def _id(obj):
return obj
# decorator that assigns test levels to test cases (classes and methods)
def level(testlevel):
if unittest.level < testlevel:
return unittest.skip("test level too low.")
return _id
def parse_args():
parser = argparse.ArgumentParser()
parser.add_argument('--level', type=int, default=3)
ns, args = parser.parse_known_args(namespace=unittest)
return ns, sys.argv[:1] + args
if __name__ == "__main__":
ns, remaining_args = parse_args()
# this invokes unittest when leveltest invoked with -m flag like:
# python -m leveltest --level=2 discover --verbose
unittest.main(module=None, argv=remaining_args)
下面是如何在示例 testproject.py 文件中使用它：
import unittest
import leveltest
# This is needed before any uses of the @leveltest.level() decorator
# to parse the "--level" command argument and set the test level when
# this test file is run directly with -m
if __name__ == "__main__":
ns, remaining_args = leveltest.parse_args()
@leveltest.level(2)
class TestStringMethods(unittest.TestCase):
@leveltest.level(5)
def test_upper(self):
self.assertEqual('foo'.upper(), 'FOO')
@leveltest.level(3)
def test_isupper(self):
self.assertTrue('FOO'.isupper())
self.assertFalse('Foo'.isupper())
@leveltest.level(4)
def test_split(self):
s = 'hello world'
self.assertEqual(s.split(), ['hello', 'world'])
# check that s.split fails when the separator is not a string
with self.assertRaises(TypeError):
s.split(2)
if __name__ == '__main__':
# this invokes unittest when this file is executed with -m
unittest.main(argv=remaining_args)
然后您可以 运行 直接通过 运行ning testproject.py 进行测试，例如：
~roottwo\projects> python testproject.py --level 2 -v
test_isupper (__main__.TestStringMethods) ... skipped 'test level too low.'
test_split (__main__.TestStringMethods) ... skipped 'test level too low.'
test_upper (__main__.TestStringMethods) ... skipped 'test level too low.'
----------------------------------------------------------------------
Ran 3 tests in 0.000s
OK (skipped=3)
~roottwo\projects> python testproject.py --level 3 -v
test_isupper (__main__.TestStringMethods) ... ok
test_split (__main__.TestStringMethods) ... skipped 'test level too low.'
test_upper (__main__.TestStringMethods) ... skipped 'test level too low.'
----------------------------------------------------------------------
Ran 3 tests in 0.001s
OK (skipped=2)
~roottwo\projects> python testproject.py --level 4 -v
test_isupper (__main__.TestStringMethods) ... ok
test_split (__main__.TestStringMethods) ... ok
test_upper (__main__.TestStringMethods) ... skipped 'test level too low.'
----------------------------------------------------------------------
Ran 3 tests in 0.001s
OK (skipped=1)
~roottwo\projects> python testproject.py --level 5 -v
test_isupper (__main__.TestStringMethods) ... ok
test_split (__main__.TestStringMethods) ... ok
test_upper (__main__.TestStringMethods) ... ok
----------------------------------------------------------------------
Ran 3 tests in 0.001s
OK
通过像这样使用单元测试发现：
~roottwo\projects> python -m leveltest --level 2 -v
test_isupper (testproject.TestStringMethods) ... skipped 'test level too low.'
test_split (testproject.TestStringMethods) ... skipped 'test level too low.'
test_upper (testproject.TestStringMethods) ... skipped 'test level too low.'
----------------------------------------------------------------------
Ran 3 tests in 0.003s
OK (skipped=3)
~roottwo\projects> python -m leveltest --level 3 discover -v
test_isupper (testproject.TestStringMethods) ... ok
test_split (testproject.TestStringMethods) ... skipped 'test level too low.'
test_upper (testproject.TestStringMethods) ... skipped 'test level too low.'
----------------------------------------------------------------------
Ran 3 tests in 0.001s
OK (skipped=2)
~roottwo\projects> python -m leveltest --level 4 -v
test_isupper (testproject.TestStringMethods) ... ok
test_split (testproject.TestStringMethods) ... ok
test_upper (testproject.TestStringMethods) ... skipped 'test level too low.'
----------------------------------------------------------------------
Ran 3 tests in 0.001s
OK (skipped=1)
~roottwo\projects> python -m leveltest discover --level 5 -v
test_isupper (testproject.TestStringMethods) ... ok
test_split (testproject.TestStringMethods) ... ok
test_upper (testproject.TestStringMethods) ... ok
----------------------------------------------------------------------
Ran 3 tests in 0.001s
OK
或者将测试用例指定为运行，如：
~roottwo\projects>python -m leveltest --level 3 testproject -v
test_isupper (testproject.TestStringMethods) ... ok
test_split (testproject.TestStringMethods) ... skipped 'test level too low.'
test_upper (testproject.TestStringMethods) ... skipped 'test level too low.'
----------------------------------------------------------------------
Ran 3 tests in 0.002s
OK (skipped=2)