如何为所有测试测试定义一个设置功能?

时间:2012-09-14 15:37:09

标签: python nosetests

我正在使用带有python的谷歌应用引擎,并希望使用nosetest运行一些测试。 我希望每个测试都运行相同的设置功能。我已经进行了很多测试,所以我不想全部考虑它们并复制和粘贴相同的功能。我可以定义一个设置函数,每个测试会先运行它吗?

感谢。

1 个答案:

答案 0 :(得分:3)

您可以编写设置功能并使用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():
    ...

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

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