diff options
| author | Vincent Rouvreau <10407034+VincentRouvreau@users.noreply.github.com> | 2020-03-16 19:16:34 +0100 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2020-03-16 19:16:34 +0100 |
| commit | f40161072b8f74f68b0ff67b6ef2be7abebec950 (patch) | |
| tree | 1b72b9bd381f52e578b1a8c889f288bbeb4d5efe /src/python/test/test_rips_complex.py | |
| parent | 592d58a95dbe5851c9f4ded9e8740c1e7a9c1502 (diff) | |
| parent | 6ed2a97421a223b4ebe31b91f48d779c2209f470 (diff) | |
Merge pull request #211 from VincentRouvreau/iterator_over_simplex_tree
Iterator over simplex tree
Diffstat (limited to 'src/python/test/test_rips_complex.py')
| -rwxr-xr-x | src/python/test/test_rips_complex.py | 6 |
1 files changed, 4 insertions, 2 deletions
diff --git a/src/python/test/test_rips_complex.py b/src/python/test/test_rips_complex.py index b02a68e1..b86e7498 100755 --- a/src/python/test/test_rips_complex.py +++ b/src/python/test/test_rips_complex.py @@ -32,7 +32,7 @@ def test_rips_from_points(): assert simplex_tree.num_simplices() == 10 assert simplex_tree.num_vertices() == 4 - assert simplex_tree.get_filtration() == [ + assert list(simplex_tree.get_filtration()) == [ ([0], 0.0), ([1], 0.0), ([2], 0.0), @@ -44,6 +44,7 @@ def test_rips_from_points(): ([1, 2], 1.4142135623730951), ([0, 3], 1.4142135623730951), ] + assert simplex_tree.get_star([0]) == [ ([0], 0.0), ([0, 1], 1.0), @@ -95,7 +96,7 @@ def test_rips_from_distance_matrix(): assert simplex_tree.num_simplices() == 10 assert simplex_tree.num_vertices() == 4 - assert simplex_tree.get_filtration() == [ + assert list(simplex_tree.get_filtration()) == [ ([0], 0.0), ([1], 0.0), ([2], 0.0), @@ -107,6 +108,7 @@ def test_rips_from_distance_matrix(): ([1, 2], 1.4142135623730951), ([0, 3], 1.4142135623730951), ] + assert simplex_tree.get_star([0]) == [ ([0], 0.0), ([0, 1], 1.0), |