Replace includes and pi in unit test. (#105)

lisitsyn · May 24, 2024 · d60b9dc · d60b9dc
1 parent 4c2f59e
commit d60b9dc
Showing 1 changed file with 4 additions and 7 deletions.
diff --git a/test/unit/data.hpp b/test/unit/data.hpp
@@ -1,14 +1,11 @@
-#include <cmath>
+#include <tapkee/defines/types.hpp>
 
-#include <fstream>
-
-#ifndef M_PI
-#define M_PI (3.14159265358979323846)
-#endif
+#include <numbers>
 
 tapkee::DenseMatrix swissroll(int N)
 {
-    tapkee::DenseVector tt = (3.0 * M_PI / 4.0) * (tapkee::DenseVector::Random(N).array() + 0.5);
+    tapkee::DenseVector tt = (3.0 * std::numbers::pi_v<double> / 4.0) *
+                              (tapkee::DenseVector::Random(N).array() + 0.5);
     tapkee::DenseVector height = tapkee::DenseVector::Random(N).array() - 0.5;
     tapkee::DenseMatrix X(N, 3);
     X.col(0) = tt.array() * tt.array().cos();