From 5056cf3b1626a04879f637d5bce9af61080e8950 Mon Sep 17 00:00:00 2001 From: vrouvrea Date: Tue, 10 May 2016 07:42:33 +0000 Subject: 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 --- src/cython/test/Simplex_tree_UT.py | 19 ++++++++++++++++++- 1 file changed, 18 insertions(+), 1 deletion(-) (limited to 'src/cython/test') 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 -- cgit v1.2.3