test_io.py 文件源码

def test_write_metrics1():
    filename = abspath(join(testdir, 'test_write_metrics1.csv'))
    if isfile(filename):
        os.remove(filename)

    metrics = pd.DataFrame({'metric1' : pd.Series([1.], index=[pd.datetime(2016,1,1)])})
    pecos.io.write_metrics(filename, metrics)
    assert_true(isfile(filename))

    from_file1 = pd.read_csv(filename)
    assert_equals(from_file1.shape, (1,2))

    # append another date
    metrics = pd.DataFrame({'metric1' : pd.Series([2.], index=[pd.datetime(2016,1,2)])})
    pecos.io.write_metrics(filename, metrics)

    from_file2 = pd.read_csv(filename)
    assert_equals(from_file2.shape, (2,2))

    # append another metric
    metrics = pd.DataFrame({'metric2' : pd.Series([3.], index=[pd.datetime(2016,1,2)])})
    pecos.io.write_metrics(filename, metrics)

    from_file3= pd.read_csv(filename)
    assert_equals(from_file3.shape, (2,3))