diff options
Diffstat (limited to 'src/cython/test/test_mini_simplex_tree.py')
-rwxr-xr-x | src/cython/test/test_mini_simplex_tree.py | 49 |
1 files changed, 22 insertions, 27 deletions
diff --git a/src/cython/test/test_mini_simplex_tree.py b/src/cython/test/test_mini_simplex_tree.py index ec1f5044..e44bfdbc 100755 --- a/src/cython/test/test_mini_simplex_tree.py +++ b/src/cython/test/test_mini_simplex_tree.py @@ -1,6 +1,4 @@ -import unittest - -import gudhi +from gudhi import MiniSimplexTree """This file is part of the Gudhi Library. The Gudhi library (Geometric Understanding in Higher Dimensions) is a generic C++ @@ -29,27 +27,24 @@ __copyright__ = "Copyright (C) 2016 INRIA" __license__ = "GPL v3" -class TestMiniSimplexTree(unittest.TestCase): - - def test_mini(self): - triangle012 = [0, 1, 2] - edge03 = [0, 3] - mini_st = gudhi.MiniSimplexTree() - self.assertTrue(mini_st.insert(triangle012)) - self.assertTrue(mini_st.insert(edge03)) - # FIXME: Remove this line - mini_st.set_dimension(2) - - edge02 = [0, 2] - 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() +def test_mini(): + triangle012 = [0, 1, 2] + edge03 = [0, 3] + mini_st = MiniSimplexTree() + assert mini_st.__is_defined() == True + assert mini_st.__is_persistence_defined() == False + assert mini_st.insert(triangle012) == True + assert mini_st.insert(edge03) == True + # FIXME: Remove this line + mini_st.set_dimension(2) + + edge02 = [0, 2] + assert mini_st.find(edge02) == True + assert mini_st.get_coface_tree(edge02, 1) == \ + [([0, 1, 2], 0.0)] + + # remove_maximal_simplex test + assert mini_st.get_coface_tree(triangle012, 1) == [] + mini_st.remove_maximal_simplex(triangle012) + assert mini_st.find(edge02) == True + assert mini_st.find(triangle012) == False |