summaryrefslogtreecommitdiff
path: root/test
diff options
context:
space:
mode:
authorievred <ievgen.redko@univ-st-etienne.fr>2020-04-08 10:08:47 +0200
committerievred <ievgen.redko@univ-st-etienne.fr>2020-04-08 10:08:47 +0200
commitc68b52d1623683e86555484bf9a4875a66957bb6 (patch)
treee7727a19ed7ba3a47a1df1ec893d0bc27c2eec57 /test
parent2c9f992157844d6253a302905417e86580ac6b12 (diff)
remove laplace from jcpot
Diffstat (limited to 'test')
-rw-r--r--test/test_da.py54
1 files changed, 0 insertions, 54 deletions
diff --git a/test/test_da.py b/test/test_da.py
index 4eaf193..1517cec 100644
--- a/test/test_da.py
+++ b/test/test_da.py
@@ -601,57 +601,3 @@ def test_jcpot_transport_class():
# check that the oos method is working
assert_equal(transp_Xs_new.shape, Xs_new.shape)
-
-
-def test_emd_laplace_class():
- """test_emd_laplace_transport
- """
- ns = 150
- nt = 200
-
- Xs, ys = make_data_classif('3gauss', ns)
- Xt, yt = make_data_classif('3gauss2', nt)
-
- otda = ot.da.EMDLaplaceTransport(reg_lap=0.01, max_iter=1000, tol=1e-9, verbose=False, log=True)
-
- # test its computed
- otda.fit(Xs=Xs, ys=ys, Xt=Xt)
-
- assert hasattr(otda, "coupling_")
- assert hasattr(otda, "log_")
-
- # test dimensions of coupling
- assert_equal(otda.coupling_.shape, ((Xs.shape[0], Xt.shape[0])))
-
- # test all margin constraints
- mu_s = unif(ns)
- mu_t = unif(nt)
-
- assert_allclose(
- np.sum(otda.coupling_, axis=0), mu_t, rtol=1e-3, atol=1e-3)
- assert_allclose(
- np.sum(otda.coupling_, axis=1), mu_s, rtol=1e-3, atol=1e-3)
-
- # test transform
- transp_Xs = otda.transform(Xs=Xs)
- [assert_equal(x.shape, y.shape) for x, y in zip(transp_Xs, Xs)]
-
- Xs_new, _ = make_data_classif('3gauss', ns + 1)
- transp_Xs_new = otda.transform(Xs_new)
-
- # check that the oos method is working
- assert_equal(transp_Xs_new.shape, Xs_new.shape)
-
- # test inverse transform
- transp_Xt = otda.inverse_transform(Xt=Xt)
- assert_equal(transp_Xt.shape, Xt.shape)
-
- Xt_new, _ = make_data_classif('3gauss2', nt + 1)
- transp_Xt_new = otda.inverse_transform(Xt=Xt_new)
-
- # check that the oos method is working
- assert_equal(transp_Xt_new.shape, Xt_new.shape)
-
- # test fit_transform
- transp_Xs = otda.fit_transform(Xs=Xs, Xt=Xt)
- assert_equal(transp_Xs.shape, Xs.shape)