Reworking from AoS to SoA for data handling

2 files changed, 35 insertions, 0 deletions

diff --git a/tests/descriptor.cpp b/tests/descriptor.cpp
index 5e23ebc..a8337e6 100644
--- a/tests/descriptor.cpp
+++ b/tests/descriptor.cpp
@@ -19,6 +19,11 @@ void check_opposite_dirs(){
 	}
 }
 
+SAW_TEST("Opposites and Dirs D1Q3"){
+	using namespace kel;
+	check_opposite_dirs<lbm::sch::Descriptor<1,3>>();
+}
+
 SAW_TEST("Opposites and Dirs D2Q5"){
 	using namespace kel;
 	check_opposite_dirs<lbm::sch::Descriptor<2,5>>();
diff --git a/tests/equilibrium.cpp b/tests/equilibrium.cpp
new file mode 100644
index 0000000..88542d2
--- /dev/null
+++ b/tests/equilibrium.cpp
@@ -0,0 +1,30 @@
+#include <forstio/test/suite.hpp>
+
+#include "../c++/equilibrium.hpp"
+
+
+namespace {
+
+template<typename Descriptor>
+void check_equilibrium(){
+	using namespace kel;
+
+	using dfi = df_info<lbm::sch::Float64,Descriptor>;
+
+	saw::data<lbm::sch::Float64> rho{1.0};
+	saw::data<lbm::sch::FixedArray<lbm::sch::Float64,>> vel;
+	for(saw::data<lbm::sch::UInt64> i{0u}; i < {Descriptor::D}; ++i){
+		vel.at(i) = {0.0};
+	}
+	auto eq = lbm::equilibrium<lbm::sch::Float64,Descriptor>(rho,vel);
+
+	for(saw::data<lbm::sch::UInt64> i{0u}; i < {Descriptor::Q}; ++i){
+		SAW_CHECK(eq(i) == dfi::weights[i.get()], "No velocity and normalized rho should be exactly the weights");
+	}
+}
+
+SAW_TEST("Equilibrium At rest D1Q3"){
+	using namespace kel;
+	check_equilibrium<lbm::sch::Descriptor<1,3>>();
+}
+}