def METHOD_NAME():
    """
    Test discretize when a 1d function is passed.
    """
    def f(x):
        return x**2
    y = discretize_model(f, (-5, 6))
    assert_allclose(y, np.arange(-5, 6) ** 2)