diff options
author | Vincent Rouvreau <10407034+VincentRouvreau@users.noreply.github.com> | 2019-09-10 11:23:03 +0200 |
---|---|---|
committer | GitHub <noreply@github.com> | 2019-09-10 11:23:03 +0200 |
commit | 9d5ddcac38d43c9df813c0ca4a37b7dc48c68913 (patch) | |
tree | d6e893623f5c621a2bd356bc053d7c0d71aa9234 /src/python/test/test_witness_complex.py | |
parent | f510a7e607b46ba8cc118cd787ff9b0b8bff091f (diff) | |
parent | 2a313b489f1412f7e9d65681cea622d88828cba3 (diff) |
Merge pull request #91 from VincentRouvreau/split_gudhi_python_in_modules
Split gudhi python in modules
Diffstat (limited to 'src/python/test/test_witness_complex.py')
-rwxr-xr-x | src/python/test/test_witness_complex.py | 62 |
1 files changed, 62 insertions, 0 deletions
diff --git a/src/python/test/test_witness_complex.py b/src/python/test/test_witness_complex.py new file mode 100755 index 00000000..36ced635 --- /dev/null +++ b/src/python/test/test_witness_complex.py @@ -0,0 +1,62 @@ +from gudhi import WitnessComplex, StrongWitnessComplex, SimplexTree + +""" This file is part of the Gudhi Library - https://gudhi.inria.fr/ - which is released under MIT. + See file LICENSE or go to https://gudhi.inria.fr/licensing/ for full license details. + Author(s): Vincent Rouvreau + + Copyright (C) 2016 Inria + + Modification(s): + - YYYY/MM Author: Description of the modification +""" + +__author__ = "Vincent Rouvreau" +__copyright__ = "Copyright (C) 2016 Inria" +__license__ = "MIT" + + +def test_empty_witness_complex(): + witness = WitnessComplex() + assert witness.__is_defined() == False + + +def test_witness_complex(): + nearest_landmark_table = [ + [[0, 0], [1, 1], [2, 2], [3, 3], [4, 4]], + [[1, 0], [2, 1], [3, 2], [4, 3], [0, 4]], + [[2, 0], [3, 1], [4, 2], [0, 3], [1, 4]], + [[3, 0], [4, 1], [0, 2], [1, 3], [2, 4]], + [[4, 0], [0, 1], [1, 2], [2, 3], [3, 4]], + ] + + witness_complex = WitnessComplex(nearest_landmark_table=nearest_landmark_table) + simplex_tree = witness_complex.create_simplex_tree(max_alpha_square=4.1) + assert simplex_tree.num_vertices() == 5 + assert simplex_tree.num_simplices() == 31 + simplex_tree = witness_complex.create_simplex_tree( + max_alpha_square=4.1, limit_dimension=2 + ) + assert simplex_tree.num_vertices() == 5 + assert simplex_tree.num_simplices() == 25 + + +def test_strong_witness_complex(): + nearest_landmark_table = [ + [[0, 0], [1, 1], [2, 2], [3, 3], [4, 4]], + [[1, 0], [2, 1], [3, 2], [4, 3], [0, 4]], + [[2, 0], [3, 1], [4, 2], [0, 3], [1, 4]], + [[3, 0], [4, 1], [0, 2], [1, 3], [2, 4]], + [[4, 0], [0, 1], [1, 2], [2, 3], [3, 4]], + ] + + strong_witness_complex = StrongWitnessComplex( + nearest_landmark_table=nearest_landmark_table + ) + simplex_tree = strong_witness_complex.create_simplex_tree(max_alpha_square=4.1) + assert simplex_tree.num_vertices() == 5 + assert simplex_tree.num_simplices() == 31 + simplex_tree = strong_witness_complex.create_simplex_tree( + max_alpha_square=4.1, limit_dimension=2 + ) + assert simplex_tree.num_vertices() == 5 + assert simplex_tree.num_simplices() == 25 |