如何在 pytest 测试中包含一个函数

Question

我正在为数值优化器编写测试。 为了测试它，我需要为它提供一个目标函数来最小化，如下所示

import pytest
from optimizer import OptimizerClass

def test_optimizer():

    def foo(params):
        return[(params[0]+2)**2]

    optimizer = OptimizerClass(objective_function = foo)
    solution = optimizer.minimize()

    assert abs(solution - (-2)) <= 10 ** -6

1. 我该怎么做才能使其有效并遵循样式指南？
2. 如果我正在编写多个都需要optimizer对象实例化的测试函数，我应该在每个函数内实例化optimizer对象还是在外部实例化一次？ 我知道两者都是可能的，但我不知道根据 Python 风格指南哪一个是正确的。

Answer 1

广告 1. - 您的解决方案看起来已经不错了，是不是有些东西不起作用了？

ad 2. - 由于优化器应该有内部状态，我认为最好为每个测试使用一个新的。 查看 pytest Fixtures。 例子：

 @pytest.fixture
 def optimizer():
     def foo(params):
         #...
     return OptimizerClass(objective_function = foo)

 def test_1(optimizer):
     # pytest will automatically resolve the argument by name and pass the function result into the test.
     result = optimizer.minimize()
     assert ...