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
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
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
def test_load_testing_data_columns_cases():
df = load_ncdhhs_data(in_file2, headers2)
tmp = list(df.columns)
assert tmp == headers2
def test_load_testing_data_shape_cases():
df = load_ncdhhs_data(in_file2, headers2)
assert df.shape == (326, 4)
def test_load_testing_data_columns_testing():
df = load_ncdhhs_data(in_file1, headers1)
tmp = list(df.columns)
assert tmp == headers1
def test_load_testing_data_shape_testing():
df = load_ncdhhs_data(in_file1, headers1)
assert df.shape == (326, 4)