def METHOD_NAME(self):
    # result shape should match non-aggregated input dims.  Lazy data should
    # be preserved.
    data = np.empty((2, 3, 4))
    axis = 1
    self.stat_func.return_value = da.arange(8)
    expected = np.arange(8).reshape(2, 4)
    wrapped_stat_func = _axis_to_single_trailing(self.stat_func)
    result = wrapped_stat_func(data, axis=axis)
    self.assertTrue(is_lazy_data(result))
    self.assertArrayEqual(as_concrete_data(result), expected)