additional python bindings for eclipse_grid

This commit is contained in:
Torbjørn Skille
2020-01-05 18:10:09 +01:00
parent c2b4d30be5
commit 4136868592
2 changed files with 127 additions and 2 deletions

View File

@@ -1,5 +1,6 @@
import unittest
import opm.io
import numpy as np
from opm.io.parser import Parser
from opm.io.ecl_state import EclipseState, test_field_props
@@ -61,17 +62,66 @@ class TestProps(unittest.TestCase):
self.assertClose(darcy, perm)
def test_volume(self):
exp = 293.3 * 293.3 * 30 # cubicfeet = 73 078.6084 cubic meter
exp *= (12*0.0254)**3 # cubic feet to cubic meter
grid = self.spe3.grid()
for i in range(grid.nx):
for j in range(grid.ny):
for k in range(grid.nz):
g_idx = grid.globalIndex(i,j,k)
exp = 293.3 * 293.3 * 30 # cubicfeet = 73 078.6084 cubic meter
exp *= (12*0.0254)**3 # cubic feet to cubic meter
if k == 0:
self.assertClose(exp, grid.getCellVolume(g_idx))
self.assertEqual(grid.getCellVolume(g_idx), grid.getCellVolume(i, j, k))
celVol1 = grid.getCellVolume()
self.assertTrue(isinstance(celVol1, np.ndarray))
self.assertEqual(celVol1.dtype, "float64")
self.assertEqual(len(celVol1), grid.nx*grid.ny*grid.nz)
mask = [0] * (grid.nx*grid.ny*grid.nz)
for ind in [0,10,11,12,15]:
mask[ind]=1
self.assertEqual(len(mask), grid.nx*grid.ny*grid.nz)
self.assertEqual(sum(mask), 5)
celVol2 = grid.getCellVolume(mask)
self.assertEqual(len(celVol2), grid.nx*grid.ny*grid.nz)
self.assertClose(exp, sum(celVol2)/5.0)
def test_depth(self):
refDepth = np.array([7330.0, 7360.0, 7400.0, 7450.0], dtype="float64")
grid = self.spe3.grid()
for i in range(grid.nx):
for j in range(grid.ny):
for k in range(grid.nz):
g_idx = grid.globalIndex(i,j,k)
exp = refDepth[k] * 12*0.0254
g_idx = grid.globalIndex(i,j,k)
self.assertClose(exp, grid.getCellDepth(g_idx))
self.assertClose(exp, grid.getCellDepth(i, j, k))
depth1 = grid.getCellDepth()
self.assertTrue(isinstance(depth1, np.ndarray))
self.assertEqual(depth1.dtype, "float64")
self.assertEqual(len(depth1), grid.nx*grid.ny*grid.nz)
mask = [0] * (grid.nx*grid.ny*grid.nz)
for k in range(grid.nz):
mask[grid.globalIndex(0, 0, k)]=1
self.assertEqual(len(mask), grid.nx*grid.ny*grid.nz)
self.assertEqual(sum(mask), grid.nz)
depth2 = grid.getCellDepth(mask)
self.assertEqual(len(depth2), grid.nx*grid.ny*grid.nz)
self.assertClose(sum(refDepth)*12*0.0254, sum(depth2))
if __name__ == "__main__":
unittest.main()