summaryrefslogtreecommitdiff
path: root/src/python/test/test_dtm.py
diff options
context:
space:
mode:
authorMarc Glisse <marc.glisse@inria.fr>2020-04-11 18:18:14 +0200
committerMarc Glisse <marc.glisse@inria.fr>2020-04-11 18:18:14 +0200
commitf9a933862050ca95b3a96d7a8572d62f7f2205a9 (patch)
tree65b1a3df0d67ea0323942cbf4bc15e6371665793 /src/python/test/test_dtm.py
parent0a404547afec2e43dd5edf9410ff079d156d691a (diff)
Use longer names
Diffstat (limited to 'src/python/test/test_dtm.py')
-rwxr-xr-xsrc/python/test/test_dtm.py18
1 files changed, 9 insertions, 9 deletions
diff --git a/src/python/test/test_dtm.py b/src/python/test/test_dtm.py
index 93b13e1a..37934fdb 100755
--- a/src/python/test/test_dtm.py
+++ b/src/python/test/test_dtm.py
@@ -8,7 +8,7 @@
- YYYY/MM Author: Description of the modification
"""
-from gudhi.point_cloud.dtm import DTM
+from gudhi.point_cloud.dtm import DistanceToMeasure
import numpy
import pytest
@@ -16,35 +16,35 @@ import pytest
def test_dtm_compare_euclidean():
pts = numpy.random.rand(1000, 4)
k = 3
- dtm = DTM(k, implementation="ckdtree")
+ dtm = DistanceToMeasure(k, implementation="ckdtree")
r0 = dtm.fit_transform(pts)
- dtm = DTM(k, implementation="sklearn")
+ dtm = DistanceToMeasure(k, implementation="sklearn")
r1 = dtm.fit_transform(pts)
assert r1 == pytest.approx(r0)
- dtm = DTM(k, implementation="sklearn", algorithm="brute")
+ dtm = DistanceToMeasure(k, implementation="sklearn", algorithm="brute")
r2 = dtm.fit_transform(pts)
assert r2 == pytest.approx(r0)
- dtm = DTM(k, implementation="hnsw")
+ dtm = DistanceToMeasure(k, implementation="hnsw")
r3 = dtm.fit_transform(pts)
assert r3 == pytest.approx(r0)
from scipy.spatial.distance import cdist
d = cdist(pts, pts)
- dtm = DTM(k, metric="precomputed")
+ dtm = DistanceToMeasure(k, metric="precomputed")
r4 = dtm.fit_transform(d)
assert r4 == pytest.approx(r0)
- dtm = DTM(k, implementation="keops")
+ dtm = DistanceToMeasure(k, implementation="keops")
r5 = dtm.fit_transform(pts)
assert r5 == pytest.approx(r0)
def test_dtm_precomputed():
dist = numpy.array([[1.0, 3, 8], [1, 5, 5], [0, 2, 3]])
- dtm = DTM(2, q=1, metric="neighbors")
+ dtm = DistanceToMeasure(2, q=1, metric="neighbors")
r = dtm.fit_transform(dist)
assert r == pytest.approx([2.0, 3, 1])
dist = numpy.array([[2.0, 2], [0, 1], [3, 4]])
- dtm = DTM(2, q=2, metric="neighbors")
+ dtm = DistanceToMeasure(2, q=2, metric="neighbors")
r = dtm.fit_transform(dist)
assert r == pytest.approx([2.0, 0.707, 3.5355], rel=0.01)