diff options
author | ROUVREAU Vincent <vincent.rouvreau@inria.fr> | 2019-06-27 21:51:22 +0200 |
---|---|---|
committer | ROUVREAU Vincent <vincent.rouvreau@inria.fr> | 2019-06-27 21:51:22 +0200 |
commit | 6a91777aa7229536d177430aadc4d7e37274cdcb (patch) | |
tree | 2ee056688826f72bedc36699b869c8dede58b332 /src/cython/test/test_reader_utils.py | |
parent | 68fadb4ebfe5b35a2eec99a9babafbee940a1b42 (diff) | |
parent | 9f249e001089af45186aaf0d196d6300705eee31 (diff) |
Merge branch 'master' of github.com:GUDHI/gudhi-devel
Diffstat (limited to 'src/cython/test/test_reader_utils.py')
-rwxr-xr-x | src/cython/test/test_reader_utils.py | 19 |
1 files changed, 10 insertions, 9 deletions
diff --git a/src/cython/test/test_reader_utils.py b/src/cython/test/test_reader_utils.py index b240c84f..36e927b0 100755 --- a/src/cython/test/test_reader_utils.py +++ b/src/cython/test/test_reader_utils.py @@ -1,4 +1,5 @@ import gudhi +import numpy as np """This file is part of the Gudhi Library. The Gudhi library (Geometric Understanding in Higher Dimensions) is a generic C++ @@ -53,7 +54,7 @@ def test_non_existing_persistence_file(): persistence = gudhi.read_persistence_intervals_grouped_by_dimension(persistence_file='pouetpouettralala.toubiloubabdou') assert persistence == [] persistence = gudhi.read_persistence_intervals_in_dimension(persistence_file='pouetpouettralala.toubiloubabdou', only_this_dim=1) - assert persistence == [] + np.testing.assert_array_equal(persistence, []) def test_read_persistence_intervals_without_dimension(): # Create test file @@ -61,11 +62,11 @@ def test_read_persistence_intervals_without_dimension(): test_file.write('# Simple persistence diagram without dimension\n2.7 3.7\n9.6 14.\n34.2 34.974\n3. inf') test_file.close() persistence = gudhi.read_persistence_intervals_in_dimension(persistence_file='persistence_intervals_without_dimension.pers') - assert persistence == [(2.7, 3.7), (9.6, 14.), (34.2, 34.974), (3., float('Inf'))] + np.testing.assert_array_equal(persistence, [(2.7, 3.7), (9.6, 14.), (34.2, 34.974), (3., float('Inf'))]) persistence = gudhi.read_persistence_intervals_in_dimension(persistence_file='persistence_intervals_without_dimension.pers', only_this_dim=0) - assert persistence == [] + np.testing.assert_array_equal(persistence, []) persistence = gudhi.read_persistence_intervals_in_dimension(persistence_file='persistence_intervals_without_dimension.pers', only_this_dim=1) - assert persistence == [] + np.testing.assert_array_equal(persistence, []) persistence = gudhi.read_persistence_intervals_grouped_by_dimension(persistence_file='persistence_intervals_without_dimension.pers') assert persistence == {-1: [(2.7, 3.7), (9.6, 14.0), (34.2, 34.974), (3.0, float('Inf'))]} @@ -75,14 +76,14 @@ def test_read_persistence_intervals_with_dimension(): test_file.write('# Simple persistence diagram with dimension\n0 2.7 3.7\n1 9.6 14.\n3 34.2 34.974\n1 3. inf') test_file.close() persistence = gudhi.read_persistence_intervals_in_dimension(persistence_file='persistence_intervals_with_dimension.pers') - assert persistence == [(2.7, 3.7), (9.6, 14.), (34.2, 34.974), (3., float('Inf'))] + np.testing.assert_array_equal(persistence, [(2.7, 3.7), (9.6, 14.), (34.2, 34.974), (3., float('Inf'))]) persistence = gudhi.read_persistence_intervals_in_dimension(persistence_file='persistence_intervals_with_dimension.pers', only_this_dim=0) - assert persistence == [(2.7, 3.7)] + np.testing.assert_array_equal(persistence, [(2.7, 3.7)]) persistence = gudhi.read_persistence_intervals_in_dimension(persistence_file='persistence_intervals_with_dimension.pers', only_this_dim=1) - assert persistence == [(9.6, 14.), (3., float('Inf'))] + np.testing.assert_array_equal(persistence, [(9.6, 14.), (3., float('Inf'))]) persistence = gudhi.read_persistence_intervals_in_dimension(persistence_file='persistence_intervals_with_dimension.pers', only_this_dim=2) - assert persistence == [] + np.testing.assert_array_equal(persistence, []) persistence = gudhi.read_persistence_intervals_in_dimension(persistence_file='persistence_intervals_with_dimension.pers', only_this_dim=3) - assert persistence == [(34.2, 34.974)] + np.testing.assert_array_equal(persistence, [(34.2, 34.974)]) persistence = gudhi.read_persistence_intervals_grouped_by_dimension(persistence_file='persistence_intervals_with_dimension.pers') assert persistence == {0: [(2.7, 3.7)], 1: [(9.6, 14.0), (3.0, float('Inf'))], 3: [(34.2, 34.974)]} |