unit-testing - 如何对网格进行单元测试显示在使用 matplotlib 完成的图中

Question

我有一种用 matplotlib 创建绘图的方法。是否应根据作为参数传递的布尔值显示或不显示网格。有没有办法在单元测试中只测试情节的这一部分，以确保它正常工作？有没有办法获得绘图定义并看到网格“打开”？

Answer 1

您可以检查any()网格线是否可见（或者all()如果愿意）：

>>> fig, ax = plt.subplots()
>>> ax.grid(True) # enable all gridlines
>>> any(line.get_visible() for line in ax.get_xgridlines() + ax.get_ygridlines())
True

>>> ax.grid(False) # disable all gridlines
>>> any(line.get_visible() for line in ax.get_xgridlines() + ax.get_ygridlines())
False

>>> ax.grid(False)
>>> ax.get_xgridlines()[0].set_visible(True) # enable 1 gridline
>>> any(line.get_visible() for line in ax.get_xgridlines() + ax.get_ygridlines())
True