def test_read_persistence_intervals_without_dimension(): # Create test file test_file = open("persistence_intervals_without_dimension.pers", "w") 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") np.testing.assert_array_equal(persistence, [(2.7, 3.7), (9.6, 14.0), (34.2, 34.974), (3.0, float("Inf"))]) persistence = gudhi.read_persistence_intervals_in_dimension( persistence_file="persistence_intervals_without_dimension.pers", only_this_dim=0) np.testing.assert_array_equal(persistence, []) persistence = gudhi.read_persistence_intervals_in_dimension( persistence_file="persistence_intervals_without_dimension.pers", only_this_dim=1) 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"))] }
def test_non_existing_persistence_file(): # Try to open a non existing 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) np.testing.assert_array_equal(persistence, [])