python - 创建绘图的 Pytest 测试函数

Question

我有几个创建绘图的函数，我在 Jupyter 笔记本中使用它们来可视化数据。

我想为这些创建基本测试，检查它们是否仍然运行，如果我进行更改，则不会在各种输入上出错。但是，如果我使用 pytest 调用这些函数，创建绘图会导致程序挂起，直到我手动最小化绘图。

import pytest 
import matplotlib.pyplot as plt 

def plot_fn():
    plt.plot([1,2,3])
    plt.show()


def test_plot_fn():
   plot_fn()

如何使用 Pytest 测试“plot_fn”等函数的运行而不出错？我尝试了以下方法，但它不起作用，我认为是因为 plt.show() 导致脚本挂起，因此无法到达 plt.close('all')。

def test_plot_fn():
   plot_fn()
   plt.close('all')

我很高兴更改绘图函数的行为，例如返回 plt 对象？

Answer 1

我不确定matplotlib如何与pytest交互，但似乎您需要使用fixtures才能实现这一点。您还需要在测试中创建某种 assert 语句，该语句将向测试发出信号，以优雅地拆除您的装置。像这样的事情应该达到预期的结果。

import pytest 
import matplotlib.pyplot as plt 

@pytest.fixture(scope='function')
def plot_fn():
    def _plot(points):
        plt.plot(points)
        yield plt.show()
        plt.close('all')
    return _plot


def test_plot_fn(plot_fn):
    points = [1, 2, 3]
    plot_fn(points)
    assert True

如果您想简单地 monkeypatch show 行为，我会像下面所示的那样进行操作。

def test_plot_fn(monkeypatch):
    monkeypatch.setattr(plt, 'show', lambda: None)
    plot_fn()