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)