test_io.py 文件源码

def test_write_monitoring_report2():# with test results and graphics (encoded and linked)
    filename1 = abspath(join(testdir, 'test_write_monitoring_report2_linked_graphics.html'))
    filename2 = abspath(join(testdir, 'test_write_monitoring_report2_encoded_graphics.html'))
    graphics_filename = abspath(join(testdir, 'custom_graphic.png'))
    if isfile(filename1):
        os.remove(filename1)
    if isfile(filename2):
        os.remove(filename2)
    if isfile(graphics_filename):
        os.remove(graphics_filename)

    pecos.logger.initialize()
    logger = logging.getLogger('pecos')

    pm = pecos.monitoring.PerformanceMonitoring()
    periods = 5
    index = pd.date_range('1/1/2016', periods=periods, freq='H')
    data = np.array([[1,2,3], [4,5,6], [7,8,9], [10,11,12], [13,14,15]])
    df = pd.DataFrame(data=data, index=index, columns=['A', 'B', 'C'])
    tfilter = pd.Series(data = (df.index < index[3]), index = df.index)
    pm.add_dataframe(df, 'test', True)
    pm.add_time_filter(tfilter)    
    pm.check_range([0,7]) # 2 test failures

    filename_root = abspath(join(testdir, 'monitoring_report_graphic'))
    test_results_graphics = pecos.graphics.plot_test_results(filename_root, pm)

    plt.figure()
    plt.plot([1, 2, 3],[1, 2, 3])
    plt.savefig(graphics_filename, format='png')
    plt.close()
    custom_graphics = [graphics_filename]

    logger.warning('Add a note')

    pecos.io.write_monitoring_report(filename1, pm, test_results_graphics, custom_graphics, encode=False)

    assert_true(isfile(filename1))

    pecos.io.write_monitoring_report(filename2, pm, test_results_graphics, custom_graphics, encode=True)

    assert_true(isfile(filename2))