diff options
Diffstat (limited to 'test/test_da.py')
-rw-r--r-- | test/test_da.py | 8 |
1 files changed, 0 insertions, 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) |