diff options
| author | Vincent Rouvreau <10407034+VincentRouvreau@users.noreply.github.com> | 2022-02-14 11:07:30 +0100 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2022-02-14 11:07:30 +0100 |
| commit | 2a2aae065bf34cfcf8bba52695ce3ae3ca6d4048 (patch) | |
| tree | 7d3b94840c2bc099a837c5852d5301c1c2a28ffc /src/python/gudhi | |
| parent | cead99a89dcb017e2e4da803de82823b52968c1c (diff) | |
| parent | 854cb0726336013cf9faede10ba61b0c6da938d3 (diff) | |
Merge pull request #463 from VincentRouvreau/python_stree_equality_operator
Add equality operator for python Simplex tree
Diffstat (limited to 'src/python/gudhi')
| -rw-r--r-- | src/python/gudhi/simplex_tree.pxd | 1 | ||||
| -rw-r--r-- | src/python/gudhi/simplex_tree.pyx | 7 |
2 files changed, 8 insertions, 0 deletions
diff --git a/src/python/gudhi/simplex_tree.pxd b/src/python/gudhi/simplex_tree.pxd index 006a24ed..70311ead 100644 --- a/src/python/gudhi/simplex_tree.pxd +++ b/src/python/gudhi/simplex_tree.pxd @@ -65,6 +65,7 @@ cdef extern from "Simplex_tree_interface.h" namespace "Gudhi": vector[vector[pair[int, pair[double, double]]]] compute_extended_persistence_subdiagrams(vector[pair[int, pair[double, double]]] dgm, double min_persistence) nogil Simplex_tree_interface_full_featured* collapse_edges(int nb_collapse_iteration) nogil except + void reset_filtration(double filtration, int dimension) nogil + bint operator==(Simplex_tree_interface_full_featured) nogil # Iterators over Simplex tree pair[vector[int], double] get_simplex_and_filtration(Simplex_tree_simplex_handle f_simplex) nogil Simplex_tree_simplices_iterator get_simplices_iterator_begin() nogil diff --git a/src/python/gudhi/simplex_tree.pyx b/src/python/gudhi/simplex_tree.pyx index c3720936..6393343f 100644 --- a/src/python/gudhi/simplex_tree.pyx +++ b/src/python/gudhi/simplex_tree.pyx @@ -642,3 +642,10 @@ cdef class SimplexTree: self.thisptr = <intptr_t>(ptr.collapse_edges(nb_iter)) # Delete old pointer del ptr + + def __eq__(self, other:SimplexTree): + """Test for structural equality + :returns: True if the 2 simplex trees are equal, False otherwise. + :rtype: bool + """ + return dereference(self.get_ptr()) == dereference(other.get_ptr()) |