如何解决如何在pytests之间重置itertools计数?
我正在使用 itertools.count() 中的索引值创建对象
class Object:
new_index = itertools.count()
def __init__(self):
self.index = next(self.new_index)
然后我在一个文件中有多个 pytest 测试。 第一个测试运行良好,返回 0,1,2 作为我的 3 个对象的索引,但后续测试继续计数,因此测试 2 为 3,4,5 等等。 有没有办法在测试之间重置这个值,以便它总是返回 0,2
解决方法
可变的全局状态使得编写独立的单元测试变得困难。如果您重新设计代码,以便将可变状态封装在某个对象中,而不是全局的,那么您的代码将更具可测试性。我会有一个 ObjectFactory
类来封装自动递增计数器:
import itertools
class Object:
def __init__(self,index):
self.index = index
class ObjectFactory:
def __init__(self):
self.counter = itertools.count()
def create(self):
index = next(self.counter)
return Object(index)
然后你的应用程序和单元测试类应该首先创建一个工厂来创建对象;单元测试类可以在 setUp
方法中创建此工厂,以避免跨多个测试重复代码。
itertools.count
迭代器无法重置,我认为。但是,以下代码将实现您重新开始计数对象的目标。测试顺序基于测试方法名称的排序顺序:
import unittest
import itertools
class MyObject:
new_index = itertools.count()
def __init__(self):
self.index = next(self.new_index)
class TestItertoolsMethods(unittest.TestCase):
def test_a(self):
obj = MyObject()
print(obj.index,end=' ')
self.assertEqual(obj.index,"Miss")
def test_b(self):
MyObject.new_index = itertools.count()
obj = MyObject()
print(obj.index,"Miss")
def test_c(self):
obj = MyObject()
print(obj.index,1,"Miss")
if __name__ == '__main__':
unittest.main()
输出
...0 0 1
----------------------------------------------------------------------
Ran 3 tests in 0.003s
OK
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。