def test_form_tables():
reporter = Reporter(output_path_template="output/test")
data_for_report = reporter.group_merged_data_and_predictions_by_location(
expected_merged_data_and_predictions)
metrics_vals = reporter.count_metrics([
metrics.utils.get_metric('sklearn.mean_absolute_error'),
metrics.utils.get_metric('MAPE')
], data_for_report)
df3d = reporter.form_tables(metrics_vals,
splitting_features=['pred_type'],
row_features=['model', 'location'],
column_features=['metric', 'horizon'])
assert expected_df3d.keys() == df3d.keys()
for title, df2d in df3d.items():
assert (expected_df3d[title].index.values == df2d.index.values).all()
assert (
expected_df3d[title].columns.values == df2d.columns.values).all()
assert (expected_df3d[title].fillna(-1) == df2d.fillna(-1)).all().all()
