add tests for statistic interceptors

Author: ManuelLerchner

tests/interceptors/DiffusionInterceptor.cpp

@@ -0,0 +1,105 @@
+#include <gtest/gtest.h>
+
+#include <filesystem>
+#include <iostream>
+#include <memory>
+
+#include "simulation/SimulationUtils.h"
+#include "simulation/interceptors/diffusion_function/DiffusionFunctionInterceptor.h"
+
+std::vector<Particle> createParticles1() {
+    std::array<double, 3> x1 = {5.5, 4.8, 3.2};
+    std::array<double, 3> v1 = {0.2, 0.1, 0};
+
+    std::array<double, 3> x2 = {5.7, 5.2, 7.4};
+    std::array<double, 3> v2 = {-0.1, 0.2, 0};
+
+    std::array<double, 3> x3 = {4.2, 5.6, 5.8};
+    std::array<double, 3> v3 = {0.1, -0.2, 0};
+
+    std::array<double, 3> x4 = {6.3, 6.2, 4};
+    std::array<double, 3> v4 = {0.1, 0.1, 0};
+
+    return {Particle(x1, v1, 1, 0), Particle(x2, v2, 1, 0), Particle(x3, v3, 1, 0), Particle(x4, v4, 1, 0)};
+}
+
+TEST(DiffusionInterceptor, IsZeroForNonMovingParticles) {
+    Logger::logger->set_level(spdlog::level::warn);
+
+    auto particles = createParticles1();
+
+    SimulationParams params_lc = TEST_DEFAULT_PARAMS_LENNARD_JONES;
+
+    Simulation simulation_lc(particles, params_lc);
+
+    DiffusionFunctionInterceptor interceptor(1);
+
+    interceptor.onSimulationStart(simulation_lc);
+
+    interceptor(1, simulation_lc);
+
+    // check if csv file was created
+
+    std::string csv_file_name = "statistics/diffusion_function.csv";
+    std::filesystem::path csv_file_path = params_lc.output_dir_path / csv_file_name;
+
+    ASSERT_TRUE(std::filesystem::exists(csv_file_path));
+
+    std::ifstream csv_file(csv_file_path);
+
+    std::string line;
+    std::getline(csv_file, line);
+
+    ASSERT_EQ(line, "\"iteration\";\"var(t)\"");
+    std::getline(csv_file, line);
+
+    ASSERT_EQ(line, "1;0");
+
+    std::getline(csv_file, line);
+
+    ASSERT_TRUE(csv_file.eof());
+}
+
+TEST(DiffusionInterceptor, CalculatesCorrectDiffusion) {
+    Logger::logger->set_level(spdlog::level::warn);
+
+    auto particles = createParticles1();
+
+    SimulationParams params_lc = TEST_DEFAULT_PARAMS_LENNARD_JONES;
+
+    Simulation simulation_lc(particles, params_lc);
+
+    DiffusionFunctionInterceptor interceptor(1);
+
+    interceptor.onSimulationStart(simulation_lc);
+
+    for (auto& p : particles) {
+        p.setX({p.getX()[0] + 1, p.getX()[1] + 1, p.getX()[2] + 1});
+    }
+
+    Simulation simulation_lc_new(particles, params_lc);
+
+    interceptor(1, simulation_lc_new);
+
+    // check if csv file was created
+
+    std::string csv_file_name = "statistics/diffusion_function.csv";
+    std::filesystem::path csv_file_path = params_lc.output_dir_path / csv_file_name;
+
+    ASSERT_TRUE(std::filesystem::exists(csv_file_path));
+
+    std::ifstream csv_file(csv_file_path);
+
+    std::string line;
+    std::getline(csv_file, line);
+
+    ASSERT_EQ(line, "\"iteration\";\"var(t)\"");
+    std::getline(csv_file, line);
+
+    // Every particle moved by sqrt(3) therfore the variance is 3
+    ASSERT_EQ(line, "1;3");
+
+    std::getline(csv_file, line);
+
+    ASSERT_TRUE(csv_file.eof());
+}

tests/interceptors/RadialDistributionFunctionInterceptor.cpp

@@ -0,0 +1,114 @@
+#include "simulation/interceptors/radial_distribution_function/RadialDistributionFunctionInterceptor.h"
+
+#include <gtest/gtest.h>
+
+#include <filesystem>
+#include <iostream>
+#include <memory>
+
+#include "simulation/SimulationUtils.h"
+
+TEST(RadialDistributionFunction, IsCorrectForSmallGrid_OneBin) {
+    Logger::logger->set_level(spdlog::level::warn);
+
+    std::vector<Particle> particles = {
+        Particle({0, 0, 0}, {0, 0, 0}, 1, 0),
+        Particle({1, 0, 0}, {0, 0, 0}, 1, 0),
+        Particle({0, 1, 0}, {0, 0, 0}, 1, 0),
+        Particle({1, 1, 0}, {0, 0, 0}, 1, 0),
+    };
+
+    SimulationParams params_lc = TEST_DEFAULT_PARAMS_LENNARD_JONES;
+
+    Simulation simulation_lc(particles, params_lc);
+
+    RadialDistributionFunctionInterceptor interceptor(2, 1);
+
+    interceptor.onSimulationStart(simulation_lc);
+
+    interceptor(1, simulation_lc);
+
+    // check if csv file was created
+
+    std::string csv_file_name = "statistics/radial_distribution_function.csv";
+    std::filesystem::path csv_file_path = params_lc.output_dir_path / csv_file_name;
+
+    ASSERT_TRUE(std::filesystem::exists(csv_file_path));
+
+    std::ifstream csv_file(csv_file_path);
+
+    std::string line;
+    std::getline(csv_file, line);
+
+    ASSERT_EQ(line, "\"iteration\";\"bin_index (w= 2.000000)\";\"samples\";\"local_density\"");
+
+    std::getline(csv_file, line);
+
+    // Hand calculated for the small example that all particles are in a grid
+    ASSERT_EQ(line, "0;0;6;0.179049");
+
+    std::getline(csv_file, line);
+
+    ASSERT_EQ(line, "1;0;6;0.179049");
+
+    std::getline(csv_file, line);
+
+    ASSERT_TRUE(csv_file.eof());
+}
+
+TEST(RadialDistributionFunction, IsCorrectForSmallGrid_TwoBins) {
+    Logger::logger->set_level(spdlog::level::warn);
+
+    std::vector<Particle> particles = {
+        Particle({0, 0, 0}, {0, 0, 0}, 1, 0),
+        Particle({1, 0, 0}, {0, 0, 0}, 1, 0),
+        Particle({0, 1, 0}, {0, 0, 0}, 1, 0),
+        Particle({1, 1, 0}, {0, 0, 0}, 1, 0),
+    };
+
+    SimulationParams params_lc = TEST_DEFAULT_PARAMS_LENNARD_JONES;
+
+    Simulation simulation_lc(particles, params_lc);
+
+    RadialDistributionFunctionInterceptor interceptor(1.2, 1);
+
+    interceptor.onSimulationStart(simulation_lc);
+
+    interceptor(1, simulation_lc);
+
+    // check if csv file was created
+
+    std::string csv_file_name = "statistics/radial_distribution_function.csv";
+    std::filesystem::path csv_file_path = params_lc.output_dir_path / csv_file_name;
+
+    ASSERT_TRUE(std::filesystem::exists(csv_file_path));
+
+    std::ifstream csv_file(csv_file_path);
+
+    std::string line;
+    std::getline(csv_file, line);
+
+    ASSERT_EQ(line, "\"iteration\";\"bin_index (w= 1.200000)\";\"samples\";\"local_density\"");
+
+    std::getline(csv_file, line);
+
+    // Hand calculated for the small example that all particles are in a grid
+    ASSERT_EQ(line, "0;0;4;0.552621");
+
+    std::getline(csv_file, line);
+
+    ASSERT_EQ(line, "0;1;2;0.039473");
+
+    std::getline(csv_file, line);
+
+    // Particles did not move, so the result should be the same
+    ASSERT_EQ(line, "1;0;4;0.552621");
+
+    std::getline(csv_file, line);
+
+    ASSERT_EQ(line, "1;1;2;0.039473");
+
+    std::getline(csv_file, line);
+
+    ASSERT_TRUE(csv_file.eof());
+}

tests/interceptors/VelocityProfileInterceptor.cpp

@@ -0,0 +1,110 @@
+#include "simulation/interceptors/velocity_profile/VelocityProfileInterceptor.h"
+
+#include <gtest/gtest.h>
+
+#include <filesystem>
+#include <iostream>
+#include <memory>
+
+#include "simulation/SimulationUtils.h"
+
+TEST(VelocityProfileInterceptor, TestBasicMeasurment) {
+    Logger::logger->set_level(spdlog::level::warn);
+
+    std::vector<Particle> particles = {
+        Particle({0, 0, 0}, {1, 2, 3}, 1, 0),
+        Particle({1, 0, 0}, {1, 2, 3}, 1, 0),
+        Particle({0, 1, 0}, {1, 2, 3}, 1, 0),
+        Particle({1, 1, 0}, {1, 2, 3}, 1, 0),
+    };
+
+    SimulationParams params_lc = TEST_DEFAULT_PARAMS_LENNARD_JONES;
+
+    Simulation simulation_lc(particles, params_lc);
+
+    std::pair<std::array<double, 3>, std::array<double, 3>> box =
+        std::make_pair(std::array<double, 3>{-0.5, -0.5, -0.5}, std::array<double, 3>{1.5, 1.5, 1.5});
+
+    VelocityProfileInterceptor interceptor(box, 2, 1);
+
+    interceptor.onSimulationStart(simulation_lc);
+
+    interceptor(1, simulation_lc);
+
+    // check if csv file was created
+
+    std::string csv_file_name_x = "statistics/velocity_profile_x_axis.csv";
+    std::filesystem::path csv_file_path_x = params_lc.output_dir_path / csv_file_name_x;
+
+    std::string csv_file_name_y = "statistics/velocity_profile_y_axis.csv";
+    std::filesystem::path csv_file_path_y = params_lc.output_dir_path / csv_file_name_y;
+
+    std::string csv_file_name_z = "statistics/velocity_profile_z_axis.csv";
+    std::filesystem::path csv_file_path_z = params_lc.output_dir_path / csv_file_name_z;
+
+    ASSERT_TRUE(std::filesystem::exists(csv_file_path_x));
+    ASSERT_TRUE(std::filesystem::exists(csv_file_path_y));
+    ASSERT_TRUE(std::filesystem::exists(csv_file_path_z));
+
+    std::ifstream csv_file_x(csv_file_path_x);
+    std::ifstream csv_file_y(csv_file_path_y);
+    std::ifstream csv_file_z(csv_file_path_z);
+
+    // check x. Values are trivial since every bin has the exact same particle velocities in it
+
+    std::string linex;
+    std::getline(csv_file_x, linex);
+
+    ASSERT_EQ(linex, "\"iteration\";\"bin_index (w= 1.000000)\";\"avg_velocity_x\";\"avg_velocity_y\";\"avg_velocity_z\"");
+    std::getline(csv_file_x, linex);
+
+    ASSERT_EQ(linex, "0;0;1;2;3");
+    std::getline(csv_file_x, linex);
+
+    ASSERT_EQ(linex, "0;1;1;2;3");
+    std::getline(csv_file_x, linex);
+
+    ASSERT_EQ(linex, "1;0;1;2;3");
+    std::getline(csv_file_x, linex);
+
+    ASSERT_EQ(linex, "1;1;1;2;3");
+    std::getline(csv_file_x, linex);
+
+    ASSERT_TRUE(csv_file_x.eof());
+
+    // check y. Values are the same as in x
+    std::string liney;
+    std::getline(csv_file_y, liney);
+
+    ASSERT_EQ(liney, "\"iteration\";\"bin_index (w= 1.000000)\";\"avg_velocity_x\";\"avg_velocity_y\";\"avg_velocity_z\"");
+    std::getline(csv_file_y, liney);
+
+    ASSERT_EQ(liney, "0;0;1;2;3");
+    std::getline(csv_file_y, liney);
+
+    ASSERT_EQ(liney, "0;1;1;2;3");
+    std::getline(csv_file_y, liney);
+
+    ASSERT_EQ(liney, "1;0;1;2;3");
+    std::getline(csv_file_y, liney);
+
+    ASSERT_EQ(liney, "1;1;1;2;3");
+    std::getline(csv_file_y, liney);
+
+    ASSERT_TRUE(csv_file_y.eof());
+
+    // check z. Values are the same as in x
+    std::string linez;
+    std::getline(csv_file_z, linez);
+
+    ASSERT_EQ(linez, "\"iteration\";\"bin_index (w= 1.000000)\";\"avg_velocity_x\";\"avg_velocity_y\";\"avg_velocity_z\"");
+    std::getline(csv_file_z, linez);
+
+    ASSERT_EQ(linez, "0;0;1;2;3");
+    std::getline(csv_file_z, linez);
+
+    ASSERT_EQ(linez, "1;0;1;2;3");
+    std::getline(csv_file_z, linez);
+
+    ASSERT_TRUE(csv_file_z.eof());
+}