python - 如何在 Python 中编写 unittest 来测试“savefig()”

Question

我想编写一个单元测试来检查 savefig(outfile) 方法是否使用我为“outfile”指定的输入名称保存我的图形。Savefig(outfile) 是我课堂上的一个方法。到目前为止，我已经编写了这段代码，但我不知道如何确保它有效

def test_savepng():
    n = 100
    example = ChaosGame(n=3)
    example.iterate(n)
    example.plot()
    example.savepng('test')
    assert example.savepng('test') == 'test.png'

该方法还检查 outfile 中是否包含“.png”，如果没有，则将其附加到末尾。