Function test_backend

test/batch/test_solve_batch.py:136–145

Check that all gradient methods run without error.

(nx)

def test_backend(nx):
    """Check that all gradient methods run without error."""
    batchsize = 2
    n = 4
    d = 2
    X = np.random.randn(batchsize, n, d)
    X = nx.from_numpy(X)
    M = dist_batch(X, X)
    solve_batch(M, reg=0.1, max_iter=10, tol=1e-5)
    solve_sample_batch(X, X, reg=0.1, max_iter=10, tol=1e-5)