Function test_fun_to_numpy

test/test_utils.py:736–751

(nx)

def test_fun_to_numpy(nx):
    arr = np.arange(5)
    arrb = nx.from_numpy(arr)

    def fun(x):  # backend function
        return nx.sum(x)

    fun_numpy = ot.utils.fun_to_numpy(fun, arrb, nx, warn=True)

    res = nx.to_numpy(fun(arrb))
    res_np = fun_numpy(arr)

    np.testing.assert_allclose(res, res_np)

    with pytest.raises(ValueError):
        ot.utils.fun_to_numpy(fun, None, nx, warn=True)