mirror of
https://github.com/OPM/ResInsight.git
synced 2025-02-25 18:55:39 -06:00
4c2a70fa40
Avoid resize if data is already present Improve functions for getting grid data and accessing single cell values Add example for read and write of grid data
61 lines
2.2 KiB
Python
61 lines
2.2 KiB
Python
import sys
|
|
import os
|
|
import math
|
|
|
|
sys.path.insert(1, os.path.join(sys.path[0], "../../"))
|
|
import rips
|
|
|
|
import dataroot
|
|
|
|
|
|
def check_corner(actual, expected):
|
|
assert math.isclose(actual.x, expected[0], abs_tol=0.1)
|
|
assert math.isclose(actual.y, expected[1], abs_tol=0.1)
|
|
assert math.isclose(actual.z, expected[2], abs_tol=0.1)
|
|
|
|
|
|
def test_10k(rips_instance, initialize_test):
|
|
casePath = dataroot.PATH + "/TEST10K_FLT_LGR_NNC/TEST10K_FLT_LGR_NNC.EGRID"
|
|
case = rips_instance.project.load_case(path=casePath)
|
|
assert len(case.grids()) == 2
|
|
grid = case.grid(index=0)
|
|
dimensions = grid.dimensions()
|
|
assert dimensions.i == 90
|
|
assert dimensions.j == 96
|
|
assert dimensions.k == 36
|
|
|
|
cell_centers = grid.cell_centers()
|
|
assert len(cell_centers) == (dimensions.i * dimensions.j * dimensions.k)
|
|
|
|
property_data_index = grid.property_data_index_from_ijk(31, 53, 21)
|
|
assert property_data_index == 177510
|
|
|
|
# Test a specific cell (results from ResInsight UI)
|
|
cell_index = 168143
|
|
assert math.isclose(3627.17, cell_centers[cell_index].x, abs_tol=0.1)
|
|
assert math.isclose(5209.75, cell_centers[cell_index].y, abs_tol=0.1)
|
|
assert math.isclose(4179.6, cell_centers[cell_index].z, abs_tol=0.1)
|
|
|
|
cell_corners = grid.cell_corners()
|
|
assert len(cell_corners) == (dimensions.i * dimensions.j * dimensions.k)
|
|
|
|
# Expected values from ResInsight UI
|
|
expected_corners = [
|
|
[3565.22, 5179.02, 4177.18],
|
|
[3655.67, 5145.34, 4176.63],
|
|
[3690.07, 5240.69, 4180.02],
|
|
[3599.87, 5275.16, 4179.32],
|
|
[3564.13, 5178.61, 4179.75],
|
|
[3654.78, 5144.79, 4179.23],
|
|
[3688.99, 5239.88, 4182.7],
|
|
[3598.62, 5274.48, 4181.96],
|
|
]
|
|
check_corner(cell_corners[cell_index].c0, expected_corners[0])
|
|
check_corner(cell_corners[cell_index].c1, expected_corners[1])
|
|
check_corner(cell_corners[cell_index].c2, expected_corners[2])
|
|
check_corner(cell_corners[cell_index].c3, expected_corners[3])
|
|
check_corner(cell_corners[cell_index].c4, expected_corners[4])
|
|
check_corner(cell_corners[cell_index].c5, expected_corners[5])
|
|
check_corner(cell_corners[cell_index].c6, expected_corners[6])
|
|
check_corner(cell_corners[cell_index].c7, expected_corners[7])
|