d11db08084
for some of these files this is needed to make to keep it compiling after the next patch because the new ErrorMacros.hpp file will no longer implicitly includes <iostream>. for the remaining files it is just good style. While at it, the includes for most of these files have been ordered in order of decreasing abstraction level.
160 lines
5.4 KiB
C++
160 lines
5.4 KiB
C++
|
|
#include <config.h>
|
|
|
|
#if HAVE_DYNAMIC_BOOST_TEST
|
|
#define BOOST_TEST_DYN_LINK
|
|
#endif
|
|
#define NVERBOSE // to suppress our messages when throwing
|
|
|
|
#define BOOST_TEST_MODULE ColumnExtractTest
|
|
#include <boost/test/unit_test.hpp>
|
|
#include <opm/core/grid/ColumnExtract.hpp>
|
|
#include <opm/core/grid/GridManager.hpp>
|
|
#include <opm/core/io/eclipse/EclipseGridParser.hpp>
|
|
|
|
#include <cstddef>
|
|
#include <iostream>
|
|
|
|
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);
|
|
|
|
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[0].begin(), columns[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);
|
|
|
|
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[i].begin(), columns[i].end());
|
|
}
|
|
}
|
|
|
|
|
|
|
|
BOOST_AUTO_TEST_CASE(DisjointColumn)
|
|
{
|
|
std::string grdecl =
|
|
"SPECGRID \n"
|
|
"3 3 3 1 F \n"
|
|
"/ \n"
|
|
" \n"
|
|
"COORD \n"
|
|
"0.0 0 0 0.0 0 3.0 \n"
|
|
"1.0 0 0 1.0 0 3.0 \n"
|
|
"2.0 0 0 2.0 0 3.0 \n"
|
|
"3.0 0 0 3.0 0 3.0 \n"
|
|
" \n"
|
|
"0.0 1.0 0 0.0 1.0 3.0 \n"
|
|
"1.0 1.0 0 1.0 1.0 3.0 \n"
|
|
"2.0 1.0 0 2.0 1.0 3.0 \n"
|
|
"3.0 1.0 0 3.0 1.0 3.0 \n"
|
|
" \n"
|
|
"0.0 2.0 0 0.0 2.0 3.0 \n"
|
|
"1.0 2.0 0 1.0 2.0 3.0 \n"
|
|
"2.0 2.0 0 2.0 2.0 3.0 \n"
|
|
"3.0 2.0 0 3.0 2.0 3.0 \n"
|
|
" \n"
|
|
"0.0 3.0 0 0.0 3.0 3.0 \n"
|
|
"1.0 3.0 0 1.0 3.0 3.0 \n"
|
|
"2.0 3.0 0 2.0 3.0 3.0 \n"
|
|
"3.0 3.0 0 3.0 3.0 3.0 \n"
|
|
"/ \n"
|
|
" \n"
|
|
"ZCORN \n"
|
|
"36*0.0 \n"
|
|
"72*1.0 \n"
|
|
"72*2.0 \n"
|
|
"36*3.0 \n"
|
|
"/ \n"
|
|
" \n"
|
|
"ACTNUM \n"
|
|
"13*1 \n"
|
|
"0 \n"
|
|
"13*1 \n"
|
|
"/ \n"
|
|
"\n";
|
|
|
|
typedef std::vector< std::vector<int> > VVI;
|
|
|
|
const int correct[][3] = { { 0, 9, 17 } , // 0
|
|
{ 1, 10, 18 } , // 1
|
|
{ 2, 11, 19 } , // 2
|
|
{ 3, 12, 20 } , // 3
|
|
{ 21, - 1, - 1 } , // 4
|
|
{ 5, 13, 22 } , // 5
|
|
{ 6, 14, 23 } , // 6
|
|
{ 7, 15, 24 } , // 7
|
|
{ 8, 16, 25 } , // 8
|
|
{ 4, - 1, - 1 } }; // 9
|
|
|
|
const std::size_t ncol = (sizeof correct) / (sizeof correct[0]);
|
|
|
|
VVI correct_answer(ncol);
|
|
|
|
for (std::size_t i = 0; i < ncol; ++i) {
|
|
for (std::size_t j = 0; j < 3; ++j) {
|
|
correct_answer[i].push_back(correct[i][j]);
|
|
}
|
|
}
|
|
correct_answer[4].resize(1);
|
|
correct_answer[9].resize(1);
|
|
|
|
std::istringstream is(grdecl);
|
|
|
|
Opm::EclipseGridParser deck;
|
|
deck.read(is);
|
|
Opm::GridManager manager(deck);
|
|
|
|
VVI columns;
|
|
Opm::extractColumn(*manager.c_grid(), columns);
|
|
|
|
#if 0
|
|
for (size_t i = 0; i < columns.size(); ++i) {
|
|
for (size_t j = 0; j < columns[i].size(); ++j) {
|
|
std::cout << columns[i][j] << ' ';
|
|
}
|
|
std::cout << '\n';
|
|
}
|
|
#endif
|
|
|
|
BOOST_CHECK_EQUAL(columns.size(), correct_answer.size());
|
|
|
|
for (VVI::iterator
|
|
xb = correct_answer.begin(), xe = correct_answer.end(),
|
|
cb = columns .begin();
|
|
xb != xe; ++xb, ++cb) {
|
|
|
|
BOOST_CHECK_EQUAL_COLLECTIONS((*xb).begin(), (*xb).end(),
|
|
(*cb).begin(), (*cb).end());
|
|
}
|
|
}
|