test_graphics.py 文件源码

def test_plot_heatmap2():
    filename = abspath(join(testdir, 'plot_heatmap2.png'))
    if isfile(filename):
        os.remove(filename)

    data = np.array([[1,2],[3,4]])

    plt.figure()
    pecos.graphics.plot_heatmap(data, cmap='jet', show_axis=True)
    plt.savefig(filename, format='png')
    plt.close()

    assert_true(isfile(filename))