def test_aggregate_metrics_is_correct_with_average_columns(self):
data = [
{
0: 1,
2: 4
},
{
0: 8,
2: 16
},
]
expected = {0: 4.5, 2: 20}
self.assertEqual(
Adapter._aggregate_metrics(data, average_columns=(0, )),
expected,
)
def test_aggregate_metrics_is_correct(self):
data = [
{
0: 1,
2: 4
},
{
0: 8,
2: 16
},
]
expected = {0: 9, 2: 20}
self.assertEqual(
Adapter._aggregate_metrics(data),
expected,
)
