1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
|
"""Tests for module sliced"""
# Author: Adrien Corenflos <adrien.corenflos@aalto.fi>
#
# License: MIT License
import numpy as np
import pytest
import ot
from ot.sliced import get_random_projections
def test_get_random_projections():
rng = np.random.RandomState(0)
projections = get_random_projections(1000, 50, rng)
np.testing.assert_almost_equal(np.sum(projections ** 2, 1), 1.)
def test_sliced_same_dist():
n = 100
rng = np.random.RandomState(0)
x = rng.randn(n, 2)
u = ot.utils.unif(n)
res = ot.sliced_wasserstein_distance(x, x, u, u, 10, seed=rng)
np.testing.assert_almost_equal(res, 0.)
def test_sliced_bad_shapes():
n = 100
rng = np.random.RandomState(0)
x = rng.randn(n, 2)
y = rng.randn(n, 4)
u = ot.utils.unif(n)
with pytest.raises(ValueError):
_ = ot.sliced_wasserstein_distance(x, y, u, u, 10, seed=rng)
def test_sliced_log():
n = 100
rng = np.random.RandomState(0)
x = rng.randn(n, 4)
y = rng.randn(n, 4)
u = ot.utils.unif(n)
res, log = ot.sliced_wasserstein_distance(x, y, u, u, 10, seed=rng, log=True)
assert len(log) == 2
projections = log["projections"]
projected_emds = log["projected_emds"]
assert len(projections) == len(projected_emds) == 10
for emd in projected_emds:
assert emd > 0
def test_sliced_different_dists():
n = 100
rng = np.random.RandomState(0)
x = rng.randn(n, 2)
u = ot.utils.unif(n)
y = rng.randn(n, 2)
res = ot.sliced_wasserstein_distance(x, y, u, u, 10, seed=rng)
assert res > 0.
def test_1d_sliced_equals_emd():
n = 100
m = 120
rng = np.random.RandomState(0)
x = rng.randn(n, 1)
a = rng.uniform(0, 1, n)
a /= a.sum()
y = rng.randn(m, 1)
u = ot.utils.unif(m)
res = ot.sliced_wasserstein_distance(x, y, a, u, 10, seed=42)
expected = ot.emd2_1d(x.squeeze(), y.squeeze(), a, u)
np.testing.assert_almost_equal(res ** 2, expected)
|