Function test_nearest_brenier_potential_fit

test/test_mapping.py:29–39

(nx)

@pytest.mark.skipif(nocvxpy, reason="No CVXPY available")
def test_nearest_brenier_potential_fit(nx):
    X = nx.ones((2, 2))
    phi, G, log = ot.mapping.nearest_brenier_potential_fit(X, X, its=3, log=True)
    np.testing.assert_almost_equal(
        to_numpy(G), to_numpy(X)
    )  # image of source should be close to target
    # test without log but with X_classes, a, b and other init method
    a = nx.ones(2) / 2
    ot.mapping.nearest_brenier_potential_fit(
        X, X, X_classes=nx.ones(2), a=a, b=a, its=1, init_method="target"
    )


@pytest.mark.skipif(nocvxpy, reason="No CVXPY available")