如何在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