python - 如何为所有 nosetests 测试定义一个设置函数？

Question

我正在将 google app engine 与 python 结合使用，并想使用 nosetest 运行一些测试。我希望每个测试都运行相同的设置函数。我已经进行了很多测试，所以我不想遍历所有测试并复制并粘贴相同的功能。我可以在某个地方定义一个设置函数并且每个测试都会先运行它吗？

谢谢。

Answer 1

您可以编写设置函数并使用 with_setup 装饰器应用它:

from nose.tools import with_setup


def my_setup():
   ...


@with_setup(my_setup)
def test_one():
    ...


@with_setup(my_setup)
def test_two():
    ...

如果您想对多个测试用例使用相同的设置，您可以使用类似的方法。首先创 build 置函数，然后使用装饰器将其应用于所有测试用例:

def my_setup(self):
    #do the setup for the test-case

def apply_setup(setup_func):
    def wrap(cls):
        cls.setup = setup_func
        return cls
    return wrap


@apply_setup(my_setup)
class MyTestCaseOne(unittest.TestCase):
    def test_one(self):
        ...
    def test_two(self):
        ...


@apply_setup(my_setup)
class MyTestCaseTwo(unittest.TestCase):
    def test_one(self):
        ...

或者另一种方法可以是简单地分配您的设置:

class MyTestCaseOne(unittest.TestCase):
    setup = my_setup