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
|
""" 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) 2021 Inria
Modification(s):
- YYYY/MM Author: Description of the modification
"""
import gudhi as gd
import numpy as np
import matplotlib as plt
import pytest
def test_array_handler():
diags = np.array([[1, 2], [3, 4], [5, 6]], np.float)
arr_diags = gd.persistence_graphical_tools._array_handler(diags)
for idx in range(len(diags)):
assert arr_diags[idx][0] == 0
np.testing.assert_array_equal(arr_diags[idx][1], diags[idx])
diags = [(1.0, 2.0), (3.0, 4.0), (5.0, 6.0)]
arr_diags = gd.persistence_graphical_tools._array_handler(diags)
for idx in range(len(diags)):
assert arr_diags[idx][0] == 0
assert arr_diags[idx][1] == diags[idx]
diags = [(0, (1.0, 2.0)), (0, (3.0, 4.0)), (0, (5.0, 6.0))]
assert gd.persistence_graphical_tools._array_handler(diags) == diags
def test_min_birth_max_death():
diags = [
(0, (0.0, float("inf"))),
(0, (0.0983494, float("inf"))),
(0, (0.0, 0.122545)),
(0, (0.0, 0.12047)),
(0, (0.0, 0.118398)),
(0, (0.118398, 1.0)),
(0, (0.0, 0.117908)),
(0, (0.0, 0.112307)),
(0, (0.0, 0.107535)),
(0, (0.0, 0.106382)),
]
assert gd.persistence_graphical_tools.__min_birth_max_death(diags) == (0.0, 1.0)
assert gd.persistence_graphical_tools.__min_birth_max_death(diags, band=4.0) == (0.0, 5.0)
def test_limit_min_birth_max_death():
diags = [
(0, (2.0, float("inf"))),
(0, (2.0, float("inf"))),
]
assert gd.persistence_graphical_tools.__min_birth_max_death(diags) == (2.0, 3.0)
assert gd.persistence_graphical_tools.__min_birth_max_death(diags, band=4.0) == (2.0, 6.0)
def test_limit_to_max_intervals():
diags = [
(0, (0.0, float("inf"))),
(0, (0.0983494, float("inf"))),
(0, (0.0, 0.122545)),
(0, (0.0, 0.12047)),
(0, (0.0, 0.118398)),
(0, (0.118398, 1.0)),
(0, (0.0, 0.117908)),
(0, (0.0, 0.112307)),
(0, (0.0, 0.107535)),
(0, (0.0, 0.106382)),
]
# check no warnings if max_intervals equals to the diagrams number
with pytest.warns(None) as record:
truncated_diags = gd.persistence_graphical_tools._limit_to_max_intervals(
diags, 10, key=lambda life_time: life_time[1][1] - life_time[1][0]
)
# check diagrams are not sorted
assert truncated_diags == diags
assert len(record) == 0
# check warning if max_intervals lower than the diagrams number
with pytest.warns(UserWarning) as record:
truncated_diags = gd.persistence_graphical_tools._limit_to_max_intervals(
diags, 5, key=lambda life_time: life_time[1][1] - life_time[1][0]
)
# check diagrams are truncated and sorted by life time
assert truncated_diags == [
(0, (0.0, float("inf"))),
(0, (0.0983494, float("inf"))),
(0, (0.118398, 1.0)),
(0, (0.0, 0.122545)),
(0, (0.0, 0.12047)),
]
assert len(record) == 1
def _limit_plot_persistence(function):
pplot = function(persistence=[()])
assert issubclass(type(pplot), plt.axes.SubplotBase)
pplot = function(persistence=[(0, float("inf"))])
assert issubclass(type(pplot), plt.axes.SubplotBase)
def test_limit_plot_persistence():
for function in [gd.plot_persistence_barcode, gd.plot_persistence_diagram, gd.plot_persistence_density]:
_limit_plot_persistence(function)
def _non_existing_persistence_file(function):
with pytest.raises(FileNotFoundError):
function(persistence_file="pouetpouettralala.toubiloubabdou")
def test_non_existing_persistence_file():
for function in [gd.plot_persistence_barcode, gd.plot_persistence_diagram, gd.plot_persistence_density]:
_non_existing_persistence_file(function)
|