From ce86d1476b32771d32b7e55566e7cab45bb57b3a Mon Sep 17 00:00:00 2001 From: ngayraud Date: Mon, 12 Aug 2019 17:03:08 -0400 Subject: Fix in test: no margin constraints here --- test/test_da.py | 8 -------- 1 file changed, 8 deletions(-) diff --git a/test/test_da.py b/test/test_da.py index 9efd2d9..2a5e50e 100644 --- a/test/test_da.py +++ b/test/test_da.py @@ -267,14 +267,6 @@ def test_unbalanced_sinkhorn_transport_class(): assert_equal(otda.cost_.shape, ((Xs.shape[0], Xt.shape[0]))) assert_equal(otda.coupling_.shape, ((Xs.shape[0], Xt.shape[0]))) - # test 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(transp_Xs.shape, Xs.shape) -- cgit v1.2.3