diff options
author | vrouvrea <vrouvrea@636b058d-ea47-450e-bf9e-a15bfbe3eedb> | 2016-06-23 16:06:31 +0000 |
---|---|---|
committer | vrouvrea <vrouvrea@636b058d-ea47-450e-bf9e-a15bfbe3eedb> | 2016-06-23 16:06:31 +0000 |
commit | 27f251520f1fc355e2b7d83225d44adbf6ce63c8 (patch) | |
tree | daa083328614684925b9e8e06e3457269d439728 /src/cython/test/test_witness_complex.py | |
parent | 8cb574ec24253e908960e00d950f2b319aabf793 (diff) |
Pytest unitary tests
git-svn-id: svn+ssh://scm.gforge.inria.fr/svnroot/gudhi/branches/ST_cythonize@1333 636b058d-ea47-450e-bf9e-a15bfbe3eedb
Former-commit-id: 0f0b284583267400574c2c2f644d5a99dfb6032e
Diffstat (limited to 'src/cython/test/test_witness_complex.py')
-rwxr-xr-x | src/cython/test/test_witness_complex.py | 48 |
1 files changed, 22 insertions, 26 deletions
diff --git a/src/cython/test/test_witness_complex.py b/src/cython/test/test_witness_complex.py index 2891935d..9f4a3523 100755 --- a/src/cython/test/test_witness_complex.py +++ b/src/cython/test/test_witness_complex.py @@ -1,6 +1,4 @@ -import unittest - -import gudhi +from gudhi import WitnessComplex """This file is part of the Gudhi Library. The Gudhi library (Geometric Understanding in Higher Dimensions) is a generic C++ @@ -29,31 +27,29 @@ __copyright__ = "Copyright (C) 2016 INRIA" __license__ = "GPL v3" -class TestWitnessComplex(unittest.TestCase): - - def test_witness_complex(self): - point_list = [[0, 0], [1, 0], [0, 1], [1, 1]] - witness = gudhi.WitnessComplex(points=point_list, - number_of_landmarks=10) - - # FIXME: Remove this line - witness.set_dimension(2) +def test_empty_witness_complex(): + witness = WitnessComplex() + assert witness.__is_defined() == False - self.assertEqual(witness.num_simplices(), 13) - self.assertEqual(witness.num_vertices(), 10) - witness.initialize_filtration() +def test_witness_complex(): + point_list = [[0, 0], [1, 0], [0, 1], [1, 1]] + witness = WitnessComplex(points=point_list, number_of_landmarks=10) + assert witness.__is_defined() == True - self.assertEqual(witness.get_filtered_tree(), - [([0], 0.0), ([1], 0.0), ([2], 0.0), ([0, 2], 0.0), - ([1, 2], 0.0), ([3], 0.0), ([4], 0.0), ([3, 4], 0.0), - ([5], 0.0), ([6], 0.0), ([7], 0.0), ([8], 0.0), - ([9], 0.0)]) + # FIXME: Remove this line + witness.set_dimension(2) - self.assertEqual(witness.get_coface_tree([2], 1), - [([0, 2], 0.0), ([1, 2], 0.0)]) - self.assertEqual(witness.get_star_tree([2]), - [([0, 2], 0.0), ([1, 2], 0.0), ([2], 0.0)]) + assert witness.num_simplices() == 13 + assert witness.num_vertices() == 10 + witness.initialize_filtration() + assert witness.get_filtered_tree() == \ + [([0], 0.0), ([1], 0.0), ([2], 0.0), ([0, 2], 0.0), + ([1, 2], 0.0), ([3], 0.0), ([4], 0.0), ([3, 4], 0.0), + ([5], 0.0), ([6], 0.0), ([7], 0.0), ([8], 0.0), + ([9], 0.0)] -if __name__ == '__main__': - unittest.main() + assert witness.get_coface_tree([2], 1) == \ + [([0, 2], 0.0), ([1, 2], 0.0)] + assert witness.get_star_tree([2]) == \ + [([0, 2], 0.0), ([1, 2], 0.0), ([2], 0.0)] |