summaryrefslogtreecommitdiff
path: root/src/python/test
diff options
context:
space:
mode:
authorVincent Rouvreau <10407034+VincentRouvreau@users.noreply.github.com>2022-06-07 11:26:17 +0200
committerGitHub <noreply@github.com>2022-06-07 11:26:17 +0200
commit2a33be5f580e5d868efe674471f32d1d025fb3c4 (patch)
treea2b4159405a4a59df7f973f5f2a9c0f47ce4cdbb /src/python/test
parent71beeb391cef793836e2d91598f9f942748edbb0 (diff)
parentf2e139123b15fee6e1512aac3824dd6664561cdb (diff)
Merge pull request #590 from VincentRouvreau/simplex_tree_extended_persistence_enhancement
extended_persistence uses directly get_persistent_pairs
Diffstat (limited to 'src/python/test')
-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 678bc905..54bafed5 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.)