Add image to simplex conversion code

[libstick.git] / tests / main.cc

diff --git a/tests/main.cc b/tests/main.cc
index ef4977e8eb9110a80f69939a54516188eef54078..dcf0abeeb978a5d5798e4575ec8a669ab70543d6 100644 (file)
--- a/tests/main.cc
+++ b/tests/main.cc
@@ -3,6 +3,8 @@
 
 #include "booleanmatrix.h"
 #include "simplicialcomplex.h"
+#include "persistence.h"
+#include "image.h"
 
 using namespace std;
 
@@ -15,8 +17,12 @@ int main(int argc, char* argv[]) {
 
     ts.add(auto_ptr<Test::Suite>(new boolean_matrix_TestSuite));
     ts.add(auto_ptr<Test::Suite>(new simplicial_complex_TestSuite));
+    ts.add(auto_ptr<Test::Suite>(new persistence_TestSuite));
+    ts.add(auto_ptr<Test::Suite>(new image_TestSuite));
 
     Test::TextOutput output(Test::TextOutput::Verbose);
-    return ts.run(output);
+    if (ts.run(output))
+        return EXIT_SUCCESS;
+    return EXIT_FAILURE;
 }