File: test_GroupState.cpp

package info (click to toggle)
opm-simulators 2025.10%2Bds-5
  • links: PTS, VCS
  • area: main
  • in suites: sid
  • size: 21,552 kB
  • sloc: cpp: 193,037; sh: 1,807; python: 1,704; lisp: 1,108; makefile: 31; awk: 10
file content (75 lines) | stat: -rw-r--r-- 2,191 bytes parent folder | download | duplicates (2)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
/*
  Copyright 2021 Equinor.

  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/>.
*/

#if HAVE_CONFIG_H
#include "config.h"
#endif // HAVE_CONFIG_H

#include <stdexcept>

#include <opm/simulators/wells/GroupState.hpp>
#include <opm/json/JsonObject.hpp>


#define BOOST_TEST_MODULE GroupStateTest
#include <boost/test/unit_test.hpp>

using namespace Opm;

class TestCommunicator {
public:
    void sum(const double *, std::size_t) const {}
};



BOOST_AUTO_TEST_CASE(GroupStateCreate)
{
    std::size_t num_phases{3};
    GroupState<double> gs(num_phases);

    BOOST_CHECK(!gs.has_production_rates("AGROUP"));
    BOOST_CHECK_THROW( gs.update_production_rates("AGROUP", {0}), std::exception);

    std::vector<double> rates{0,1,2};
    gs.update_production_rates("AGROUP", rates);
    BOOST_CHECK(gs.has_production_rates("AGROUP"));

    BOOST_CHECK_THROW( gs.production_rates("NO_SUCH_GROUP"), std::exception );
    auto r2 = gs.production_rates("AGROUP");
    BOOST_CHECK( r2 == rates );

    gs.update_injection_rein_rates("CGROUP", rates);


    BOOST_CHECK(!gs.has_production_control("NO_SUCH_GROUP"));
    BOOST_CHECK(!gs.has_production_control("AGROUP"));

    gs.production_control("AGROUP", Group::ProductionCMode::GRAT);
    BOOST_CHECK(gs.has_production_control("AGROUP"));
    BOOST_CHECK(gs.production_control("AGROUP") == Group::ProductionCMode::GRAT);
    BOOST_CHECK_THROW(gs.production_control("BGROUP"), std::exception);


    auto gs2 = gs;
    BOOST_CHECK(gs2 == gs);
    TestCommunicator comm;
    gs.communicate_rates(comm);
    BOOST_CHECK(gs2 == gs);
}