当输出是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/

10-13 06:43