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
|
import unittest
import gudhi
class TestRipsComplex(unittest.TestCase):
def test_rips(self):
point_list = [[0, 0], [1, 0], [0, 1], [1, 1]]
rips_complex = gudhi.RipsComplex(points=point_list, max_dimension=1,
max_edge_length=42)
self.assertEqual(rips_complex.num_simplices(), 10)
self.assertEqual(rips_complex.num_vertices(), 4)
self.assertEqual(rips_complex.get_filtered_tree(),
[([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)])
self.assertEqual(rips_complex.get_star_tree([0]),
[([0], 0.0), ([0, 1], 1.0), ([0, 2], 1.0),
([0, 3], 1.4142135623730951)])
self.assertEqual(rips_complex.get_coface_tree([0], 1),
[([0, 1], 1.0), ([0, 2], 1.0),
([0, 3], 1.4142135623730951)])
filtered_rips = gudhi.RipsComplex(points=point_list, max_dimension=1,
max_edge_length=1.0)
self.assertEqual(filtered_rips.num_simplices(), 8)
self.assertEqual(filtered_rips.num_vertices(), 4)
if __name__ == '__main__':
unittest.main()
|