1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
|
""" 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
"""
from gudhi import RipsComplex
from math import sqrt
__author__ = "Vincent Rouvreau"
__copyright__ = "Copyright (C) 2016 Inria"
__license__ = "MIT"
def test_empty_rips():
rips_complex = RipsComplex()
def test_rips_from_points():
point_list = [[0, 0], [1, 0], [0, 1], [1, 1]]
rips_complex = RipsComplex(points=point_list, max_edge_length=42)
simplex_tree = rips_complex.create_simplex_tree(max_dimension=1)
assert simplex_tree.__is_defined() == True
assert simplex_tree.__is_persistence_defined() == False
assert simplex_tree.num_simplices() == 10
assert simplex_tree.num_vertices() == 4
assert list(simplex_tree.get_filtration()) == [
([0], 0.0),
([1], 0.0),
([2], 0.0),
([3], 0.0),
([0, 1], 1.0),
([0, 2], 1.0),
([1, 3], 1.0),
([2, 3], 1.0),
([1, 2], 1.4142135623730951),
([0, 3], 1.4142135623730951),
]
assert simplex_tree.get_star([0]) == [
([0], 0.0),
([0, 1], 1.0),
([0, 2], 1.0),
([0, 3], 1.4142135623730951),
]
assert simplex_tree.get_cofaces([0], 1) == [
([0, 1], 1.0),
([0, 2], 1.0),
([0, 3], 1.4142135623730951),
]
def test_filtered_rips_from_points():
point_list = [[0, 0], [1, 0], [0, 1], [1, 1]]
filtered_rips = RipsComplex(points=point_list, max_edge_length=1.0)
simplex_tree = filtered_rips.create_simplex_tree(max_dimension=1)
assert simplex_tree.__is_defined() == True
assert simplex_tree.__is_persistence_defined() == False
assert simplex_tree.num_simplices() == 8
assert simplex_tree.num_vertices() == 4
def test_sparse_filtered_rips_from_points():
point_list = [[0, 0], [1, 0], [0, 1], [1, 1]]
filtered_rips = RipsComplex(points=point_list, max_edge_length=1.0, sparse=0.001)
simplex_tree = filtered_rips.create_simplex_tree(max_dimension=1)
assert simplex_tree.__is_defined() == True
assert simplex_tree.__is_persistence_defined() == False
assert simplex_tree.num_simplices() == 8
assert simplex_tree.num_vertices() == 4
def test_rips_from_distance_matrix():
distance_matrix = [[0], [1, 0], [1, sqrt(2), 0], [sqrt(2), 1, 1, 0]]
rips_complex = RipsComplex(distance_matrix=distance_matrix, max_edge_length=42)
simplex_tree = rips_complex.create_simplex_tree(max_dimension=1)
assert simplex_tree.__is_defined() == True
assert simplex_tree.__is_persistence_defined() == False
assert simplex_tree.num_simplices() == 10
assert simplex_tree.num_vertices() == 4
assert list(simplex_tree.get_filtration()) == [
([0], 0.0),
([1], 0.0),
([2], 0.0),
([3], 0.0),
([0, 1], 1.0),
([0, 2], 1.0),
([1, 3], 1.0),
([2, 3], 1.0),
([1, 2], 1.4142135623730951),
([0, 3], 1.4142135623730951),
]
assert simplex_tree.get_star([0]) == [
([0], 0.0),
([0, 1], 1.0),
([0, 2], 1.0),
([0, 3], 1.4142135623730951),
]
assert simplex_tree.get_cofaces([0], 1) == [
([0, 1], 1.0),
([0, 2], 1.0),
([0, 3], 1.4142135623730951),
]
def test_filtered_rips_from_distance_matrix():
distance_matrix = [[0], [1, 0], [1, sqrt(2), 0], [sqrt(2), 1, 1, 0]]
filtered_rips = RipsComplex(distance_matrix=distance_matrix, max_edge_length=1.0)
simplex_tree = filtered_rips.create_simplex_tree(max_dimension=1)
assert simplex_tree.__is_defined() == True
assert simplex_tree.__is_persistence_defined() == False
assert simplex_tree.num_simplices() == 8
assert simplex_tree.num_vertices() == 4
|