summaryrefslogtreecommitdiff
path: root/src/python/test/test_simplex_tree.py
diff options
context:
space:
mode:
authorVincent Rouvreau <vincent.rouvreau@inria.fr>2022-03-21 17:47:59 +0100
committerVincent Rouvreau <vincent.rouvreau@inria.fr>2022-03-21 17:47:59 +0100
commit17dc48527dcc8ee7e5eab95f9fdde3e236f4ad47 (patch)
treeeac09d9815c037f06146bb5c76c8b34487e35cda /src/python/test/test_simplex_tree.py
parentbbff86f1218fc7bc9976353901aa94cfa54792f6 (diff)
extended_persistence uses directly get_persistent_pairs
Diffstat (limited to 'src/python/test/test_simplex_tree.py')
-rwxr-xr-xsrc/python/test/test_simplex_tree.py4
1 files changed, 4 insertions, 0 deletions
diff --git a/src/python/test/test_simplex_tree.py b/src/python/test/test_simplex_tree.py
index a8180ce8..23458eb2 100755
--- a/src/python/test/test_simplex_tree.py
+++ b/src/python/test/test_simplex_tree.py
@@ -320,6 +320,10 @@ def test_extend_filtration():
]
dgms = st.extended_persistence(min_persistence=-1.)
+ assert len(dgms) == 4
+ # Sort by (death-birth) descending - we are only interested in those with the longest life span
+ for idx in range(4):
+ dgms[idx] = sorted(dgms[idx], key=lambda x:(-abs(x[1][0]-x[1][1])))
assert dgms[0][0][1][0] == pytest.approx(2.)
assert dgms[0][0][1][1] == pytest.approx(3.)