Use boost test

1 files changed, 10 insertions, 9 deletions

diff --git a/testsuite.cpp b/testsuite.cpp
index 09ac565..cc51cfa 100644
--- a/testsuite.cpp
+++ b/testsuite.cpp
@@ -1,4 +1,9 @@
-// FFT Test
+#define BOOST_TEST_MODULE fft_test
+
+#include <boost/test/included/unit_test.hpp>
+#include <boost/test/data/dataset.hpp>
+#include <boost/test/data/monomorphic.hpp>
+#include <boost/test/data/test_case.hpp>
 
 #include "autocorrelation.h"
 #include "fft.h"
@@ -162,13 +167,10 @@ public:
 								    [](const std::complex<double>& x, const std::complex<double>&y) -> double
 								    { return abs(y - x);}
 								    );
-				if (diff > tolerance)
-					std::cout << "Error: Results diff: " << diff << "\n";
+				BOOST_CHECK_MESSAGE(diff <= tolerance, "Error: Results diff: " << diff);
 			}
 
-			if (mElapsed > mReference->mElapsed) {
-				std::cout << "Error: " << mName << " too slow!\n";
-			}
+			BOOST_CHECK_MESSAGE(mElapsed <= mReference->mElapsed, "Error: " << mName << " too slow!");
 		}
 	}
 
@@ -249,7 +251,8 @@ public:
 	}
 };
 
-int main(int argc, char* argv[]) {
+BOOST_AUTO_TEST_CASE(fft)
+{
 	std::vector<std::complex<double>> v(4096, 0);
 
 	generate(v);
@@ -279,8 +282,6 @@ int main(int argc, char* argv[]) {
 
 	MeasureTuner_RR measureTuner_RR(v);
 	measureTuner_RR.run();
-
-	return 0;
 }
 
 // TODO: