如果setUp在Python unittest中引发异常,则停止运行测试

Question

我有这个测试类:

class mytest(unittest.TestCase):
    def setUp(self):
        os.mkdir(...)
        ...

    def tearDown(self):
        shutil.rmtree(...)

    def test_one(self):
        ...

    def test_two(self):
        ...

如果事情失败后,mkdir运行的时候已经跑了setUp的test_one,它仍然会尝试运行setUp的test_two.此时我会收到错误,mkdir因为rmtree没有运行.

有没有办法告诉Python unittest如果setUp失败就停止运行当前的测试？我不打算停止常规的测试失败.

Answer 1

在setUp方法中添加失败调用.

def setUp(self):
    try:
        somethingThatMightFail()
    except:
        self.fail()