max/tug
1
0
Fork 0
mirror of https://git.gfz-potsdam.de/naaice/tug.git synced 2025-12-16 02:48:23 +01:00
Code Issues Packages Projects Releases Wiki Activity

feat: Add unit tests for Velocities functionality

Browse Source
This commit is contained in:
Max Lübke 2024-12-11 15:30:10 +01:00
parent 13d55f9260
commit 1a11991af0
3 changed files with 86 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
include/tug/tug.hpp
View File

@ -1,6 +1,6 @@
#pragma once #pragma once
#include <tug/Advection/Advection.hpp> // #include <tug/Advection/Advection.hpp>
#include <tug/Boundary.hpp> #include <tug/Boundary.hpp>
#include <tug/Core/Matrix.hpp> #include <tug/Core/Matrix.hpp>
#include <tug/Diffusion/Diffusion.hpp> #include <tug/Diffusion/Diffusion.hpp>

1
test/CMakeLists.txt
View File

@ -14,6 +14,7 @@ setup.cpp
testDiffusion.cpp testDiffusion.cpp
testFTCS.cpp testFTCS.cpp
testBoundary.cpp testBoundary.cpp
testVelocities.cpp
) )
target_link_libraries(testTug tug GTest::gtest) target_link_libraries(testTug tug GTest::gtest)

84
test/testVelocities.cpp Normal file
View File

@ -0,0 +1,84 @@
#include "tug/Boundary.hpp"
#include "tug/Core/Matrix.hpp"
#include <cstddef>
#include <tug/Advection/Velocities.hpp>
#include <gtest/gtest.h>
#define VELOCITIES_TEST(x) TEST(Velocities, x)
VELOCITIES_TEST(SteadyStateCenter) {
// Arrange
constexpr std::size_t rows = 25;
constexpr std::size_t cols = 25;
constexpr std::size_t center_row = rows / 2;
constexpr std::size_t center_col = cols / 2;
constexpr double K = 1E-2;
tug::RowMajMat<double> hydHeads =
tug::RowMajMat<double>::Constant(rows, cols, 1);
tug::RowMajMat<double> permKX =
tug::RowMajMat<double>::Constant(rows, cols, K);
tug::RowMajMat<double> permKY =
tug::RowMajMat<double>::Constant(rows, cols, K);
tug::Grid64 gridHeads(hydHeads);
gridHeads.setDomain(100, 100);
gridHeads.setAlpha(permKX, permKY);
tug::Boundary bcH = tug::Boundary(gridHeads);
bcH.setBoundarySideConstant(tug::BC_SIDE_LEFT, 1);
bcH.setBoundarySideConstant(tug::BC_SIDE_RIGHT, 1);
bcH.setBoundarySideConstant(tug::BC_SIDE_TOP, 1);
bcH.setBoundarySideConstant(tug::BC_SIDE_BOTTOM, 1);
bcH.setInnerBoundary(center_row, center_col, 10);
tug::Velocities<double, tug::HYDRAULIC_MODE::STEADY_STATE,
tug::HYDRAULIC_RESOLVE::EXPLICIT>
velocities(gridHeads, bcH);
velocities.run();
const auto &velocitiesX = velocities.getVelocitiesX();
const auto &velocitiesY = velocities.getVelocitiesY();
// Assert
// check velocities in x-direction
for (std::size_t i_rows = 0; i_rows < rows; i_rows++) {
for (std::size_t i_cols = 0; i_cols < cols + 1; i_cols++) {
if (i_rows <= center_row && i_cols <= center_col) {
EXPECT_LE(velocitiesX(i_rows, i_cols), 0);
} else if (i_rows > center_row && i_cols > center_col) {
EXPECT_GE(velocitiesX(i_rows, i_cols), 0);
} else if (i_rows <= center_row && i_cols > center_col) {
EXPECT_GE(velocitiesX(i_rows, i_cols), 0);
} else if (i_rows > center_row && i_cols <= center_col) {
EXPECT_LE(velocitiesX(i_rows, i_cols), 0);
} else {
FAIL() << "Uncovered case";
}
}
}
// check velocities in y-direction
for (std::size_t i_rows = 0; i_rows < rows + 1; i_rows++) {
for (std::size_t i_cols = 0; i_cols < cols; i_cols++) {
if (i_rows <= center_row && i_cols <= center_col) {
EXPECT_LE(velocitiesY(i_rows, i_cols), 0);
} else if (i_rows > center_row && i_cols > center_col) {
EXPECT_GE(velocitiesY(i_rows, i_cols), 0);
} else if (i_rows <= center_row && i_cols > center_col) {
EXPECT_LE(velocitiesY(i_rows, i_cols), 0);
} else if (i_rows > center_row && i_cols <= center_col) {
EXPECT_GE(velocitiesY(i_rows, i_cols), 0);
} else {
FAIL() << "Uncovered case";
}
}
}
}