多个副本

Question

比方说，我有一个简单的灯具，如以下（使用pytest-Django的，但我认为它应该适用于pytest以及）：

@pytest.fixture 
def my_thing(request, db): 
    thing = MyModel.objects.create() 
    request.addfinalizer(lambda: thing.delete()) 
    return thing 

这个伟大的工程时我的测试需要一个MyModel实例。但是如果我需要两个（或三个或四个）呢？我希望每个实例都是独特的，但要以相同的方式进行设置。

我可以复制/粘贴代码并重命名灯具功能，但看起来不够雅致。

同样，我曾尝试：

@pytest.fixture 
def my_thing_1(my_thing): 
    return my_thing 

@pytest.fixture 
def my_thing_2(my_thing): 
    return my_thing 

然而，这些出现的回归为MyModel的同一个实例。

有没有办法用pytest的内置功能来做我想做的事？或者，我可以将我的灯具的设置/拆卸转换为辅助函数，这样我就不会复制太多的代码。

或者我是否以错误的方式处理这件事？

Answer 1

我的方法可能会创建一个夹具，可以生成你的对象：

@pytest.fixture 
def thing(request, db): 
    class ThingFactory(object): 
     def get(self): 
      thing = MyModel.objects.create() 
      request.addfinalizer(thing.delete) 
      return thing 
    return ThingFactory() 

def test_thing(thing): 
    thing1 = thing.get() 
    thing2 = thing.get() 

很明显，你可以让.get()带参数等

（PS：另外请注意没有必要为拉姆达在终结者）

Answer 2

我很晚才到达这个问题......但是，使用参数化夹具，并简单地返回你想要复制的夹具似乎也工作。

import pytest 
import random 

@pytest.fixture 
def name(): 
    return random.randint(0, 10) 


@pytest.fixture(params=[0, 1]) 
def parametrized_name(request, name): 
    return name 


def test_something(parametrized_name): 
    print "Testing with name: " + str(parametrized_name) 

如果你运行上面的测试，你就会得到2个不同的“名字”灯具

$ pytest -s blah.py 
============================================= test session starts ============================================== 
platform linux2 -- Python 2.7.14, pytest-3.3.2, py-1.5.2, pluggy-0.6.0 
rootdir: /home/bsquizza/repos/blah/tests, inifile: 
collected 2 items                        

blah.py Testing with name: 7 
.Testing with name: 10 
.                        [100%] 

=========================================== 2 passed in 0.01 seconds ===========================================