diff options
| author | ngayraud <nat.gayraud@gmail.com> | 2019-08-12 17:03:08 -0400 |
|---|---|---|
| committer | ngayraud <nat.gayraud@gmail.com> | 2019-08-12 17:03:08 -0400 |
| commit | ce86d1476b32771d32b7e55566e7cab45bb57b3a (patch) | |
| tree | 4d5fe2d98d249a252cda18db29ff47edc472a2ab | |
| parent | 2633116175a09c468d953489c3fc7bab6aa69057 (diff) | |
Fix in test: no margin constraints here
| -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) |