diff options
Diffstat (limited to 'src/python/test')
-rw-r--r-- | src/python/test/test_persistence_graphical_tools.py | 105 |
1 files changed, 105 insertions, 0 deletions
diff --git a/src/python/test/test_persistence_graphical_tools.py b/src/python/test/test_persistence_graphical_tools.py new file mode 100644 index 00000000..1ad1ae23 --- /dev/null +++ b/src/python/test/test_persistence_graphical_tools.py @@ -0,0 +1,105 @@ +""" 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., 2.), (3., 4.), (5., 6.)] + 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., 2.)), (0, (3., 4.)), (0, (5., 6.))] + assert gd.persistence_graphical_tools._array_handler(diags) == diags + +def test_min_birth_max_death(): + diags = [ + (0, (0., float("inf"))), + (0, (0.0983494, float("inf"))), + (0, (0., 0.122545)), + (0, (0., 0.12047)), + (0, (0., 0.118398)), + (0, (0.118398, 1.)), + (0, (0., 0.117908)), + (0, (0., 0.112307)), + (0, (0., 0.107535)), + (0, (0., 0.106382)), + ] + assert gd.persistence_graphical_tools.__min_birth_max_death(diags) == (0., 1.) + assert gd.persistence_graphical_tools.__min_birth_max_death(diags, band=4.) == (0., 5.) + +def test_limit_min_birth_max_death(): + diags = [ + (0, (2., float("inf"))), + (0, (2., float("inf"))), + ] + assert gd.persistence_graphical_tools.__min_birth_max_death(diags) == (2., 3.) + assert gd.persistence_graphical_tools.__min_birth_max_death(diags, band = 4.) == (2., 6.) + +def test_limit_to_max_intervals(): + diags = [ + (0, (0., float("inf"))), + (0, (0.0983494, float("inf"))), + (0, (0., 0.122545)), + (0, (0., 0.12047)), + (0, (0., 0.118398)), + (0, (0.118398, 1.)), + (0, (0., 0.117908)), + (0, (0., 0.112307)), + (0, (0., 0.107535)), + (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., float("inf"))), + (0, (0.0983494, float("inf"))), + (0, (0.118398, 1.0)), + (0, (0., 0.122545)), + (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) |