70 lines
2.1 KiB
C++
70 lines
2.1 KiB
C++
|
|
|
|
|
|
#define BOOST_TEST_DYN_LINK
|
|
#define NVERBOSE // to suppress our messages when throwing
|
|
|
|
#define BOOST_TEST_MODULE ColumnExtractTest
|
|
#include <boost/test/unit_test.hpp>
|
|
#include <opm/core/ColumnExtract.hpp>
|
|
#include <opm/core/GridManager.hpp>
|
|
|
|
BOOST_AUTO_TEST_CASE(SingleColumnTest)
|
|
{
|
|
using namespace Opm;
|
|
|
|
const int size_x = 1, size_y = 1, size_z = 10;
|
|
GridManager manager(size_x, size_y, size_z);
|
|
|
|
// We do our own numbering
|
|
UnstructuredGrid* grid = const_cast<UnstructuredGrid*>(manager.c_grid());
|
|
grid->global_cell = (int*)malloc(sizeof(int) * size_x * size_y * size_z);
|
|
for(int i = 0; i < size_x * size_y * size_z; ++i) {
|
|
grid->global_cell[i] = i;
|
|
}
|
|
|
|
std::pair<std::vector<int>, std::vector<std::vector<int> > > columns;
|
|
extractColumn(*manager.c_grid(), columns);
|
|
|
|
std::vector<int> correct_answer;
|
|
|
|
for(int i = 0; i < 10; i++) {
|
|
correct_answer.push_back(i);
|
|
}
|
|
BOOST_CHECK_EQUAL_COLLECTIONS(correct_answer.begin(), correct_answer.end(),
|
|
columns.second[0].begin(), columns.second[0].end());
|
|
|
|
}
|
|
|
|
|
|
BOOST_AUTO_TEST_CASE(FourByFourColumnTest)
|
|
{
|
|
const int size_x = 4, size_y = 4, size_z = 10;
|
|
using namespace Opm;
|
|
GridManager manager(size_x, size_y, size_z);
|
|
|
|
// We do our own numbering
|
|
UnstructuredGrid* grid = const_cast<UnstructuredGrid*>(manager.c_grid());
|
|
grid->global_cell = (int*)malloc(sizeof(int) * size_x * size_y * size_z);
|
|
for(int i = 0; i < size_x * size_y * size_z; ++i) {
|
|
grid->global_cell[i] = i;
|
|
}
|
|
|
|
std::pair<std::vector<int>, std::vector<std::vector<int> > > columns;
|
|
extractColumn(*manager.c_grid(), columns);
|
|
|
|
std::vector<std::vector<int> > correct_answer;
|
|
correct_answer.resize(size_x * size_y);
|
|
for(int i = 0; i < size_x * size_y; i++) {
|
|
for(int j = 0; j < 10; j++) {
|
|
correct_answer[i].push_back( i + j*size_x*size_y);
|
|
}
|
|
}
|
|
|
|
for(int i = 0; i < size_x * size_y; i++) {
|
|
BOOST_CHECK_EQUAL_COLLECTIONS(correct_answer[i].begin(), correct_answer[i].end(),
|
|
columns.second[i].begin(), columns.second[i].end());
|
|
}
|
|
|
|
}
|