97 lines
2.9 KiB
C++
97 lines
2.9 KiB
C++
/*
|
|
Copyright 2019 Equinor ASA.
|
|
|
|
This file is part of the Open Porous Media project (OPM).
|
|
|
|
OPM is free software: you can redistribute it and/or modify
|
|
it under the terms of the GNU General Public License as published by
|
|
the Free Software Foundation, either version 3 of the License, or
|
|
(at your option) any later version.
|
|
|
|
OPM is distributed in the hope that it will be useful,
|
|
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
GNU General Public License for more details.
|
|
|
|
You should have received a copy of the GNU General Public License
|
|
along with OPM. If not, see <http://www.gnu.org/licenses/>.
|
|
*/
|
|
|
|
#include <stdexcept>
|
|
#include <algorithm>
|
|
|
|
#define BOOST_TEST_MODULE WLIST_TEST
|
|
|
|
#include <boost/test/unit_test.hpp>
|
|
|
|
#include <opm/parser/eclipse/EclipseState/Schedule/WList.hpp>
|
|
#include <opm/parser/eclipse/EclipseState/Schedule/WListManager.hpp>
|
|
|
|
BOOST_AUTO_TEST_CASE(CreateWLIST) {
|
|
Opm::WList wlist;
|
|
BOOST_CHECK_EQUAL(wlist.size(), 0);
|
|
wlist.add("W1");
|
|
BOOST_CHECK_EQUAL(wlist.size(), 1);
|
|
|
|
|
|
wlist.del("NO_SUCH_WELL");
|
|
BOOST_CHECK_EQUAL(wlist.size(), 1);
|
|
|
|
wlist.del("W1");
|
|
BOOST_CHECK_EQUAL(wlist.size(), 0);
|
|
|
|
wlist.add("W1");
|
|
wlist.add("W2");
|
|
wlist.add("W3");
|
|
|
|
auto wells = wlist.wells();
|
|
BOOST_CHECK_EQUAL(wells.size(), 3);
|
|
BOOST_CHECK( std::find(wells.begin(), wells.end(), "W1") != wells.end());
|
|
BOOST_CHECK( std::find(wells.begin(), wells.end(), "W2") != wells.end());
|
|
BOOST_CHECK( std::find(wells.begin(), wells.end(), "W3") != wells.end());
|
|
|
|
std::vector<std::string> wells2;
|
|
for (const auto& well : wlist)
|
|
wells2.push_back(well);
|
|
|
|
BOOST_CHECK_EQUAL(wells2.size(), 3);
|
|
BOOST_CHECK( std::find(wells2.begin(), wells2.end(), "W1") != wells2.end());
|
|
BOOST_CHECK( std::find(wells2.begin(), wells2.end(), "W2") != wells2.end());
|
|
BOOST_CHECK( std::find(wells2.begin(), wells2.end(), "W3") != wells2.end());
|
|
}
|
|
|
|
|
|
BOOST_AUTO_TEST_CASE(WLISTManager) {
|
|
Opm::WListManager wlm;
|
|
BOOST_CHECK(!wlm.hasList("NO_SUCH_LIST"));
|
|
|
|
|
|
{
|
|
auto& wlist1 = wlm.newList("LIST1");
|
|
wlist1.add("A");
|
|
wlist1.add("B");
|
|
wlist1.add("C");
|
|
}
|
|
// If a new list is added with the same name as an existing list the old
|
|
// list is dropped and a new list is created.
|
|
{
|
|
auto& wlist1 = wlm.newList("LIST1");
|
|
BOOST_CHECK_EQUAL(wlist1.size(), 0);
|
|
}
|
|
auto& wlist1 = wlm.newList("LIST1");
|
|
auto& wlist2 = wlm.newList("LIST2");
|
|
|
|
wlist1.add("W1");
|
|
wlist1.add("W2");
|
|
wlist1.add("W3");
|
|
|
|
wlist2.add("W1");
|
|
wlist2.add("W2");
|
|
wlist2.add("W3");
|
|
|
|
// The delWell operation will work across all well lists.
|
|
wlm.delWell("W1");
|
|
BOOST_CHECK( std::find(wlist1.begin(), wlist1.end(), "W1") == wlist1.end());
|
|
BOOST_CHECK( std::find(wlist2.begin(), wlist2.end(), "W1") == wlist2.end());
|
|
}
|