Python Unittest:确保对象不包含上次运行的数据

时间:2018-05-09 06:24:28

标签: python python-3.x python-unittest

我正在使用setUp从我创建的类创建一个新对象。我的理解是这个函数将在测试用例中的每个测试之前执行,并且这应该导致为每个测试创建 new 对象。似乎这不是正在发生的事情;至少不在我的测试中。

这是我的班级:

class Entity:
    def __init__(self, entities = []):
        self.entities = entities

    def add(self, entity: str):
        if entity not in self.entities:
            self.entities.append(entity)

以下是相应的测试:

import unittest
from entity import Entity

class EntityTestCase(unittest.TestCase):
    def setUp(self):
        self.entity = Entity()
        print("Starting Entites = {}".format(self.entity.entities))

    def testA(self):
        self.entity.add("Foo")
        self.assertEqual(self.entity.entities, ["Foo"])

    def testB(self):
        self.entity.add("Bar")
        self.assertEqual(self.entity.entities, ["Bar"])

if __name__ == '__main__':
    unittest.main()

我希望testAtestB以新的Entity对象开头。也就是说,我希望Entity.entities成为每个测试的全新列表。

我正在使用python -m unittest discover -v运行测试,结果如下:

$ python -m unittest discover -v
testA (test_entity.EntityTestCase) ... Starting Entites = []
ok
testB (test_entity.EntityTestCase) ... Starting Entites = ['Foo']
FAIL

======================================================================
FAIL: testB (test_entity.EntityTestCase)
----------------------------------------------------------------------
Traceback (most recent call last):
  File "/Users/julioguzman/Sites/Foobar/test_entity.py", line 15, in testB
    self.assertEqual(self.entity.entities, ["Bar"])
AssertionError: Lists differ: ['Foo', 'Bar'] != ['Bar']

First differing element 0:
'Foo'
'Bar'

First list contains 1 additional elements.
First extra element 1:
'Bar'

- ['Foo', 'Bar']
+ ['Bar']

----------------------------------------------------------------------
Ran 2 tests in 0.001s

FAILED (failures=1)

如您所见,testBtestA的数据开始。这不是理想的行为,尽管它可能是预期的行为。

我如何确保我的物品清洁"每次测试?为什么会这样?

1 个答案:

答案 0 :(得分:2)

你有

class Entity:
    def __init__(self, entities = []):
        self.entities = entities

默认参数被实例化一次,从而导致这个常见问题

只需修改__init__以使用代理值并指定一个全新的列表,如果是这样的话:

class Entity:
    def __init__(self, entities=NotImplemented):
        if entities is NotImplemented:
            self.entities = []
        else:
            self.entities = entities

顺便说一句,这就是编写单元测试的一个原因:它们还测试是否使用预期的行为正确创建了类。当setUp清楚地表明它创建了一个全新的对象时,这不是你的测试的问题 - 错误在于实现意外的事情。