OilfieldToolsNet/opm-common
4
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

test_state2.py: added test_grid.

Browse Source
This commit is contained in:
Steinar Foss 2019-09-13 11:57:30 +02:00
parent af096fddf9
commit 97728e9fda
4 changed files with 17 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
python/cxx/eclipse_grid.cpp
View File

@ -45,8 +45,11 @@ void python::common::export_EclipseGrid(py::module& module) {
py::class_< EclipseGrid >( module, "EclipseGrid")
.def( "_getXYZ", &getXYZ )
.def( "nactive", &getNumActive )
.def( "cartesianSize", &getCartesianSize )
.def_property_readonly("NX", &EclipseGrid::getNX)
.def_property_readonly("NY", &EclipseGrid::getNY)
.def_property_readonly("NZ", &EclipseGrid::getNZ)
.def_property_readonly( "nactive", &getNumActive )
.def_property_readonly( "cartesianSize", &getCartesianSize )
.def( "globalIndex", &getGlobalIndex )
.def( "getIJK", &getIJK )
.def( "_cellVolume1G", &cellVolume1G)

2
python/cxx/eclipse_state.cpp
View File

@ -93,7 +93,7 @@ void python::common::export_EclipseState(py::module& module) {
.def(py::init<const Deck&>())
.def_property_readonly( "title", &EclipseState::getTitle )
.def( "_props", &EclipseState::get3DProperties, ref_internal)
.def( "_grid", &EclipseState::getInputGrid, ref_internal)
.def( "grid", &EclipseState::getInputGrid, ref_internal)
.def( "_cfg", &EclipseState::cfg, ref_internal)
.def( "_tables", &EclipseState::getTableManager, ref_internal)
.def( "has_input_nnc", &EclipseState::hasInputNNC )

13
python/tests/test_state.py
View File

@ -53,19 +53,6 @@ SATNUM
cpa = opm.io.parse('tests/data/CORNERPOINT_ACTNUM.DATA')
cls.state = cls.spe3.state
cls.cp_state = cpa.state
def test_grid(self):
grid = self.state.grid()
self.assertTrue('EclipseGrid' in repr(grid))
self.assertEqual(9, grid.getNX())
self.assertEqual(9, grid.getNY())
self.assertEqual(4, grid.getNZ())
self.assertEqual(9*9*4, grid.nactive())
self.assertEqual(9*9*4, grid.cartesianSize())
g,i,j,k = 295,7,5,3
self.assertEqual(g, grid.globalIndex(i,j,k))
self.assertEqual((i,j,k), grid.getIJK(g))
def test_summary(self):

11
python/tests/test_state2.py
View File

@ -65,5 +65,16 @@ SATNUM
def test_state_nnc(self):
self.assertFalse(self.state.has_input_nnc())
def test_grid(self):
grid = self.state.grid()
self.assertEqual(9, grid.NX)
self.assertEqual(9, grid.NY)
self.assertEqual(4, grid.NZ)
self.assertEqual(9*9*4, grid.nactive)
self.assertEqual(9*9*4, grid.cartesianSize)
g,i,j,k = 295,7,5,3
self.assertEqual(g, grid.globalIndex(i,j,k))
self.assertEqual((i,j,k), grid.getIJK(g))