additional python bindings for eclipse_grid
This commit is contained in:
@@ -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()
|
||||
|
||||
Reference in New Issue
Block a user