From bf3a53d0d73f57fb801875a5dde11b9ede10b0aa Mon Sep 17 00:00:00 2001 From: vrouvrea Date: Fri, 17 Mar 2017 13:11:22 +0000 Subject: Euclidean strong witness complex cytonization git-svn-id: svn+ssh://scm.gforge.inria.fr/svnroot/gudhi/branches/ST_cythonize@2195 636b058d-ea47-450e-bf9e-a15bfbe3eedb Former-commit-id: de15e40b8f0efcc59b2f07ffb5f5d59ccab8d51c --- src/cython/test/test_euclidean_witness_complex.py | 34 ++++++++++++++++++++++- 1 file changed, 33 insertions(+), 1 deletion(-) (limited to 'src/cython/test') diff --git a/src/cython/test/test_euclidean_witness_complex.py b/src/cython/test/test_euclidean_witness_complex.py index cb150b7e..0947cc09 100755 --- a/src/cython/test/test_euclidean_witness_complex.py +++ b/src/cython/test/test_euclidean_witness_complex.py @@ -34,6 +34,38 @@ def test_empty_euclidean_witness_complex(): def test_witness_complex(): point_cloud = [[1.0, 1.0], [7.0, 0.0], [4.0, 6.0], [9.0, 6.0], [0.0, 14.0], [2.0, 19.0], [9.0, 17.0]] - landmarks = gudhi.pick_n_random_points(points=point_cloud,nb_points=3) + landmarks = [[1.0, 1.0], [7.0, 0.0], [4.0, 6.0]] euclidean_witness_complex = gudhi.EuclideanWitnessComplex(landmarks=landmarks, witnesses = point_cloud) simplex_tree = euclidean_witness_complex.create_simplex_tree(max_alpha_square=4.1) + + assert landmarks[0] == euclidean_witness_complex.get_point(0) + assert landmarks[1] == euclidean_witness_complex.get_point(1) + assert landmarks[2] == euclidean_witness_complex.get_point(2) + + assert simplex_tree.get_filtered_tree() == [([0], 0.0), ([1], 0.0), + ([0, 1], 0.0), ([2], 0.0), ([0, 2], 0.0), ([1, 2], 0.0), + ([0, 1, 2], 0.0)] + +def test_empty_euclidean_strong_witness_complex(): + euclidean_strong_witness = gudhi.EuclideanStrongWitnessComplex() + assert euclidean_strong_witness.__is_defined() == False + +def test_strong_witness_complex(): + point_cloud = [[1.0, 1.0], [7.0, 0.0], [4.0, 6.0], [9.0, 6.0], + [0.0, 14.0], [2.0, 19.0], [9.0, 17.0]] + landmarks = [[1.0, 1.0], [7.0, 0.0], [4.0, 6.0]] + euclidean_strong_witness_complex = gudhi.EuclideanStrongWitnessComplex(landmarks=landmarks, witnesses = point_cloud) + simplex_tree = euclidean_strong_witness_complex.create_simplex_tree(max_alpha_square=14.9) + + assert landmarks[0] == euclidean_strong_witness_complex.get_point(0) + assert landmarks[1] == euclidean_strong_witness_complex.get_point(1) + assert landmarks[2] == euclidean_strong_witness_complex.get_point(2) + + assert simplex_tree.get_filtered_tree() == [([0], 0.0), ([1], 0.0), ([2], 0.0)] + + simplex_tree = euclidean_strong_witness_complex.create_simplex_tree(max_alpha_square=100.0) + + assert simplex_tree.get_filtered_tree() == [([0], 0.0), ([1], 0.0), + ([2], 0.0), ([1, 2], 15.0), ([0, 2], 34.0), ([0, 1], 37.0), + ([0, 1, 2], 37.0)] + -- cgit v1.2.3