def METHOD_NAME(self):
    with self.assertRaises(ValueError):
        merger = AvgMerger(merged_shape=(1, 3, 2, 3))
        merger.finalize()
        merger.aggregate(torch.zeros(1, 3, 2, 2), (3, 3))