Skip to content

Commit f7f09a9

Browse files
committed
add test
1 parent 4f66eac commit f7f09a9

File tree

1 file changed

+5
-0
lines changed

1 file changed

+5
-0
lines changed

test/test_bregman.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1253,6 +1253,11 @@ def test_lazy_empirical_sinkhorn(nx):
12531253
X_sb, X_tb, 1, numIterMax=numIterMax, isLazy=True, batchSize=1, log=False
12541254
)
12551255

1256+
# test lazy plan
1257+
np.testing.assert_allclose(
1258+
G_sqe[1, 1], G_lazy[1, 1], atol=1e-03
1259+
) # metric sqeuclidian
1260+
12561261
# check constraints
12571262
np.testing.assert_allclose(
12581263
sinkhorn_sqe.sum(1), G_sqe.sum(1), atol=1e-05

0 commit comments

Comments
 (0)