当输出是matplotlib对象时,有没有办法通过doctest?我还使用doctest框架作为代码示例(不仅用于测试输出)
所以我的问题看起来像这样:
plt.plot(grid, pdf); plt.title('Random Normal 1D using Kernel1D.kde function'); plt.grid(); plt.show()
Expected nothing
Got:
[<matplotlib.lines.Line2D object at 0x00000208BDA8E2B0>]
Text(0.5, 1.0, 'Random Normal 1D using Kernel1D.kde function')
我想发生的事情是,在我绘制任何内容时都要通过doctest。谢谢。
最佳答案
您可以使用doctest.ELLIPSIS使字符串匹配任何内容。
如果要避免看到绘图并直接转到评估报告,您的代码仍会在plt.show()
中显示问题。为此,您可以使用doctest.SKIP。检查以下示例:
import matplotlib.pyplot as plt
def test():
"""
Code example:
>>> 1 + 1
2
>>> plt.plot([1, 2, 3])
[...
>>> plt.show() #doctest: +SKIP
>>> plt.close()
"""
pass
if __name__ == "__main__":
import doctest
doctest.testmod(verbose=True, optionflags=doctest.ELLIPSIS)
这将返回以下报告:
Trying:
1 + 1
Expecting:
2
ok
Trying:
plt.plot([1, 2, 3])
Expecting:
[...
ok
Trying:
plt.close()
Expecting nothing
ok
1 items had no tests:
__main__
1 items passed all tests:
3 tests in __main__.test
3 tests in 2 items.
3 passed and 0 failed.
Test passed.
关于python - 如何通过绘图传递Python doctest,我们在Stack Overflow上找到一个类似的问题:https://stackoverflow.com/questions/57249007/