diff options
| author | yuichi-ike <yuichi.ike.1990@gmail.com> | 2020-04-06 21:19:55 +0900 |
|---|---|---|
| committer | yuichi-ike <yuichi.ike.1990@gmail.com> | 2020-04-06 21:19:55 +0900 |
| commit | a4fa5f673784a842e9fac13003c843d454c888a4 (patch) | |
| tree | b412473b8707f6f593c8b836eea4ca31f5e0cbae /src/python/test | |
| parent | 15586d479be885319dde6f703c3126176b796732 (diff) | |
bug fixed, parameter name changed
Diffstat (limited to 'src/python/test')
| -rw-r--r-- | src/python/test/test_weighted_rips.py | 13 |
1 files changed, 6 insertions, 7 deletions
diff --git a/src/python/test/test_weighted_rips.py b/src/python/test/test_weighted_rips.py index f0db6798..7896fb78 100644 --- a/src/python/test/test_weighted_rips.py +++ b/src/python/test/test_weighted_rips.py @@ -1,6 +1,6 @@ """ This file is part of the Gudhi Library - https://gudhi.inria.fr/ - which is released under MIT. See file LICENSE or go to https://gudhi.inria.fr/licensing/ for full license details. - Author(s): Yuichi Ike + Author(s): Yuichi Ike and Masatoshi Takenouchi Copyright (C) 2020 Inria @@ -10,18 +10,17 @@ from gudhi.weighted_rips_complex import WeightedRipsComplex from gudhi.point_cloud.dtm import DTM -import numpy +import numpy as np from scipy.spatial.distance import cdist import pytest def test_dtm_rips_complex(): - pts = numpy.array([[2.0, 2], [0, 1], [3, 4]]) + pts = np.array([[2.0, 2], [0, 1], [3, 4]]) dist = cdist(pts,pts) dtm = DTM(2, q=2, metric="precomputed") r = dtm.fit_transform(dist) - w_rips = WeightedRipsComplex(distance_mattix=dist, filtration_values=r) + w_rips = WeightedRipsComplex(distance_mattix=dist, weights=r) st = w_rips.create_simplex_tree(max_dimension=2) - diag = st.persistence() - assert diag == [(0, (1.5811388300841898, float("inf"))), (0, (1.5811388300841898, 2.699172818834085)), (0, (1.5811388300841898, 2.699172818834085))] + persistence_intervals0 = st.persistence_intervals_in_dimension(0) + assert persistence_intervals0 == pytest.approx(np.array([[1.58113883, 2.69917282],[1.58113883, 2.69917282], [1.58113883, float("inf")]])) -
\ No newline at end of file |