2 files changed, 11 insertions, 29 deletions
diff --git a/src/Bottleneck_distance/example/CMakeLists.txt b/src/Bottleneck_distance/example/CMakeLists.txt
index 9a37f02e..dc1da31c 100644
--- a/src/Bottleneck_distance/example/CMakeLists.txt
+++ b/src/Bottleneck_distance/example/CMakeLists.txt
@@ -19,6 +19,10 @@ if (NOT CGAL_WITH_EIGEN3_VERSION VERSION_LESS 4.8.1)
       COMMAND $<TARGET_FILE:alpha_rips_persistence_bottleneck_distance>
       "${CMAKE_SOURCE_DIR}/data/points/tore3D_1307.off" "-r" "0.15" "-m" "0.12" "-d" "3" "-p" "3")
 
+  add_test(NAME Bottleneck_read_file_example
+      COMMAND $<TARGET_FILE:bottleneck_read_file_example>
+      "${CMAKE_SOURCE_DIR}/data/persistence_diagram/first.pers" "${CMAKE_SOURCE_DIR}/data/persistence_diagram/second.pers")
+
   install(TARGETS bottleneck_read_file_example DESTINATION bin)
   install(TARGETS bottleneck_basic_example DESTINATION bin)
   install(TARGETS alpha_rips_persistence_bottleneck_distance DESTINATION bin)
diff --git a/src/Bottleneck_distance/example/bottleneck_read_file_example.cpp b/src/Bottleneck_distance/example/bottleneck_read_file_example.cpp
index bde05825..238d99ad 100644
--- a/src/Bottleneck_distance/example/bottleneck_read_file_example.cpp
+++ b/src/Bottleneck_distance/example/bottleneck_read_file_example.cpp
@@ -20,9 +20,8 @@
  *    along with this program.  If not, see <http://www.gnu.org/licenses/>.
  */
 
-#define CGAL_HAS_THREADS
-
 #include <gudhi/Bottleneck.h>
+#include <gudhi/reader_utils.h>
 #include <iostream>
 #include <vector>
 #include <utility>  // for pair
@@ -30,43 +29,22 @@
 #include <sstream>
 #include <string>
 
-std::vector< std::pair<double, double> > read_diagram_from_file(const char* filename) {
-  std::ifstream in;
-  in.open(filename);
-  std::vector< std::pair<double, double> > result;
-  if (!in.is_open()) {
-    std::cerr << "File : " << filename << " do not exist. The program will now terminate \n";
-    throw "File do not exist \n";
-  }
-
-  std::string line;
-  while (!in.eof()) {
-    getline(in, line);
-    if (line.length() != 0) {
-      std::stringstream lineSS;
-      lineSS << line;
-      double beginn, endd;
-      lineSS >> beginn;
-      lineSS >> endd;
-      result.push_back(std::make_pair(beginn, endd));
-    }
-  }
-  in.close();
-  return result;
-}  // read_diagram_from_file
-
 int main(int argc, char** argv) {
   if (argc < 3) {
     std::cout << "To run this program please provide as an input two files with persistence diagrams. Each file " <<
         "should contain a birth-death pair per line. Third, optional parameter is an error bound on a bottleneck" <<
         " distance (set by default to zero). The program will now terminate \n";
+    return -1;
   }
-  std::vector< std::pair< double, double > > diag1 = read_diagram_from_file(argv[1]);
-  std::vector< std::pair< double, double > > diag2 = read_diagram_from_file(argv[2]);
+  std::vector<std::pair<double, double>> diag1 = read_persistence_intervals_in_dimension(argv[1]);
+  std::vector<std::pair<double, double>> diag2 = read_persistence_intervals_in_dimension(argv[2]);
+
   double tolerance = 0.;
   if (argc == 4) {
     tolerance = atof(argv[3]);
   }
   double b = Gudhi::persistence_diagram::bottleneck_distance(diag1, diag2, tolerance);
   std::cout << "The distance between the diagrams is : " << b << ". The tolerance is : " << tolerance << std::endl;
+
+  return 0;
 }