如何在 Python 中的每个单元测试之前和之后运行特定代码

Question

以下是我在文件中的测试结构。

类
设置
测试01
测试02
测试03
拆解

我需要在每次测试之前和之后运行特定代码。

之前，我可以从设置中调用该代码。 但是在测试之后，我不知道该怎么做。 显然，从拆卸中调用代码将适用于最后一个测试，但我怎样才能让它为中间的测试运行呢？

Answer 1

假设您正确使用了从unittest.TestCase 派生的类，则setUp 方法在每次测试之前运行，tearDown 方法在每次测试之后运行。 Check the documentation。所以把你的代码放在这两种方法中是完全可行的。