Add examples and tests for remove_maximal_leaf

git-svn-id: svn+ssh://scm.gforge.inria.fr/svnroot/gudhi/branches/ST_cythonize@1156 636b058d-ea47-450e-bf9e-a15bfbe3eedb

Former-commit-id: 7b3884709a74e19101b23f30af125f7b3365de74

1 files changed, 18 insertions, 1 deletions

diff --git a/src/cython/test/Simplex_tree_UT.py b/src/cython/test/Simplex_tree_UT.py
index 56113370..9a62e009 100755
--- a/src/cython/test/Simplex_tree_UT.py
+++ b/src/cython/test/Simplex_tree_UT.py
@@ -42,6 +42,17 @@ class TestSimplexTree(unittest.TestCase):
     self.assertEqual(st.get_skeleton_tree(1), [([0, 1], 0.0), ([0, 2], 4.0), ([0], 0.0), ([1, 2], 4.0), ([1], 0.0), ([2], 4.0)])
     self.assertEqual(st.get_skeleton_tree(0), [([0], 0.0), ([1], 0.0), ([2], 4.0)])
 
+    # remove_maximal_simplex test
+    self.assertEqual(st.get_coface_tree([0,1,2], 1), [])
+    st.remove_maximal_simplex([0,1,2])
+    self.assertEqual(st.get_skeleton_tree(2), [([0, 1], 0.0), ([0, 2], 4.0), ([0], 0.0), ([1, 2], 4.0), ([1], 0.0), ([2], 4.0)])
+    self.assertFalse(st.find([0,1,2]))
+    self.assertTrue(st.find([0,1]))
+    self.assertTrue(st.find([0,2]))
+    self.assertTrue(st.find([0]))
+    self.assertTrue(st.find([1]))
+    self.assertTrue(st.find([2]))
+
   def test_rips(self):
     rips_complex = gudhi.SimplexTree(points=[[0,0],[1,0],[0,1],[1,1]],max_dimension=1,max_edge_length=42)
 
@@ -56,7 +67,7 @@ class TestSimplexTree(unittest.TestCase):
     self.assertEqual(filtered_rips.num_simplices(), 8)
     self.assertEqual(filtered_rips.num_vertices(), 4)
 
-  def test_split(self):
+  def test_mini(self):
     triangle012 = [0,1,2]
     edge03 = [0,3]
     mini_st = gudhi.MiniSimplexTree()
@@ -69,5 +80,11 @@ class TestSimplexTree(unittest.TestCase):
     self.assertTrue(mini_st.find(edge02))
     self.assertEqual(mini_st.get_coface_tree(edge02, 1), [([0, 1, 2], 0.0)])
 
+    # remove_maximal_simplex test
+    self.assertEqual(mini_st.get_coface_tree(triangle012, 1), [])
+    mini_st.remove_maximal_simplex(triangle012)
+    self.assertTrue(mini_st.find(edge02))
+    self.assertFalse(mini_st.find(triangle012))
+
 if __name__ == '__main__':
     unittest.main()
\ No newline at end of file