python - 如何在 Python 中的单元测试设置阶段实例化对象

Question

我一直在使用 Python 进行单元测试，但无法弄清楚我应该如何在设置阶段实例化我想要测试的对象，并最终为每个测试创建一个新对象。例如，我有以下要测试的类：

class Cfg():
    data = {}

    def set(self, key, value):
        self.data[key] = value

    def get(self, key):
        return self.data.get(key, None)

对于每个单元测试，我想要一个新实例化的Cfg对象。我的测试如下所示：

from cfg import Cfg

class TestCfg():
    def setup(self):
        self.cfg = Cfg()

    def teardown(self):
        self.cfg = None

    def test_a(self):
        self.cfg.set('foo', 'bar')
        assert self.cfg.get('foo') == 'bar'

    def test_b(self):
        assert self.cfg.get('foo') == 'bar'

我不明白为什么test_b会通过。我期望setup并tearDown“重置”我的cfg实例，但它似乎cfg在测试之间持续存在。我在这里做错了什么以及如何实现预期的行为？

Answer 1

这与您如何编写Cfg课程有关。将data初始化移动到__init__方法中：

class Cfg():
    def __init__(self):
        self.data = {}

    def set(self, key, value):
        self.data[key] = value

    def get(self, key):
        return self.data.get(key, None)

而且，你会看到失败test_b。