summaryrefslogtreecommitdiff
path: root/src/cython/test/test_euclidean_witness_complex.py
diff options
context:
space:
mode:
authorROUVREAU Vincent <vincent.rouvreau@inria.fr>2019-06-26 16:59:58 +0200
committerROUVREAU Vincent <vincent.rouvreau@inria.fr>2019-06-26 16:59:58 +0200
commitfd47a2d926e78d3b83e79da3c959fb73809d09e4 (patch)
treed440223947ae58fa06d15c575f08023902bad898 /src/cython/test/test_euclidean_witness_complex.py
parent2ffd579440e6f0158588f8fa905e2f1dc49a5158 (diff)
Format Python files with black
Diffstat (limited to 'src/cython/test/test_euclidean_witness_complex.py')
-rwxr-xr-xsrc/cython/test/test_euclidean_witness_complex.py68
1 files changed, 52 insertions, 16 deletions
diff --git a/src/cython/test/test_euclidean_witness_complex.py b/src/cython/test/test_euclidean_witness_complex.py
index c8510b1e..f5eae5fa 100755
--- a/src/cython/test/test_euclidean_witness_complex.py
+++ b/src/cython/test/test_euclidean_witness_complex.py
@@ -19,31 +19,60 @@ def test_empty_euclidean_witness_complex():
euclidean_witness = gudhi.EuclideanWitnessComplex()
assert euclidean_witness.__is_defined() == False
+
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]]
+ 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_witness_complex = gudhi.EuclideanWitnessComplex(landmarks=landmarks, witnesses = point_cloud)
+ 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_filtration() == [([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)]
+ assert simplex_tree.get_filtration() == [
+ ([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]]
+ 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)
+ 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)
@@ -51,9 +80,16 @@ def test_strong_witness_complex():
assert simplex_tree.get_filtration() == [([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_filtration() == [([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)]
-
+ simplex_tree = euclidean_strong_witness_complex.create_simplex_tree(
+ max_alpha_square=100.0
+ )
+
+ assert simplex_tree.get_filtration() == [
+ ([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),
+ ]