示例#1
0
def test_dataframe_ordinary_least_squares():
    df = load_ncdhhs_data(in_file1, headers1)
    df1 = aggregate_weekly(df, 'Total')
    dataframe_ordinary_least_squares(df1,
                                     y_col_name='Total',
                                     x_col_name='Week',
                                     showplot=False)
    assert True
示例#2
0
def test_dataframe_lowess():
    df = load_ncdhhs_data(in_file1, headers1)
    df1 = aggregate_weekly(df, 'Total')
    dataframe_lowess(df1,
                     y_col_name='Total',
                     x_col_name='Week',
                     show_plot=False,
                     show_table=False)
    assert True
示例#3
0
def test_aggregate_weekly():
    df = load_ncdhhs_data(in_file1, headers1)
    df1 = aggregate_weekly(df, 'Total')
    diff = df1.index[1] - df1.index[0]
    assert diff.days == 7
示例#4
0
def test_load_testing_data_columns_cases():
    df = load_ncdhhs_data(in_file2, headers2)
    tmp = list(df.columns)
    assert tmp == headers2
示例#5
0
def test_load_testing_data_shape_cases():
    df = load_ncdhhs_data(in_file2, headers2)
    assert df.shape == (326, 4)
示例#6
0
def test_load_testing_data_columns_testing():
    df = load_ncdhhs_data(in_file1, headers1)
    tmp = list(df.columns)
    assert tmp == headers1
示例#7
0
def test_load_testing_data_shape_testing():
    df = load_ncdhhs_data(in_file1, headers1)
    assert df.shape == (326, 4)