summaryrefslogtreecommitdiff
path: root/test/test_ot.py
diff options
context:
space:
mode:
Diffstat (limited to 'test/test_ot.py')
-rw-r--r--test/test_ot.py5
1 files changed, 4 insertions, 1 deletions
diff --git a/test/test_ot.py b/test/test_ot.py
index 18b6294..c756e51 100644
--- a/test/test_ot.py
+++ b/test/test_ot.py
@@ -337,7 +337,10 @@ def test_dual_variables():
# Check that both cost computations are equivalent
np.testing.assert_almost_equal(cost1, log['cost'])
check_duality_gap(a, b, M, G, log['u'], log['v'], log['cost'])
-
+
+ viol=log['u'][:,None]+log['v'][None,:]-M
+
+ assert viol.max()<1e-8
def check_duality_gap(a, b, M, G, u, v, cost):
cost_dual = np.vdot(a, u) + np.vdot(b, v)