diff options
author | vrouvrea <vrouvrea@636b058d-ea47-450e-bf9e-a15bfbe3eedb> | 2016-05-10 07:42:33 +0000 |
---|---|---|
committer | vrouvrea <vrouvrea@636b058d-ea47-450e-bf9e-a15bfbe3eedb> | 2016-05-10 07:42:33 +0000 |
commit | 5056cf3b1626a04879f637d5bce9af61080e8950 (patch) | |
tree | 3908ef1d8c17e4ab0e5d4627ef8497a0194515b2 /src/cython/test | |
parent | d9780fc1fda42d78038c327b05203e55e3b40fd0 (diff) |
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
Diffstat (limited to 'src/cython/test')
-rwxr-xr-x | src/cython/test/Simplex_tree_UT.py | 19 |
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 |