def METHOD_NAME(func):
    grid = RasterModelGrid((3, 4), xy_spacing=(3.0, 4.0))
    z = [[0.0, 0.0, 0.0, 0.0], [0.0, 5.0, 5.0, 0.0], [0.0, 0.0, 0.0, 0.0]]
    expected = getattr(grid, func)(z)
    out = np.empty_like(expected)
    actual = getattr(grid, func)(z, out=out)
    assert actual is out
    assert_array_equal(actual, expected)