mirror of
https://github.com/OPM/ResInsight.git
synced 2026-10-26 07:00:17 -05:00
Kristian Bendiksen
ebf7a061d8
Rim3dView_exportContourMapToText exports the contour map of an Eclipse or GeoMech contour map view. The implementation is extracted from the RICF_SOURCE_INIT feature RicExportContourMapToTextFeature into a static helper returning std::expected, used by both the object method and the legacy command. The Python EclipseContourMap/GeoMechContourMap.export_to_text() wrappers are kept as aliases of the generated method.
84 lines
2.6 KiB
Python
84 lines
2.6 KiB
Python
import sys
|
|
import os
|
|
import pytest
|
|
import tempfile
|
|
|
|
sys.path.insert(1, os.path.join(sys.path[0], "../../"))
|
|
|
|
import rips
|
|
import dataroot
|
|
|
|
|
|
def test_visible_cells(rips_instance, initialize_test):
|
|
"""Test the visible_cells() method on a view"""
|
|
case_root_path = dataroot.PATH + "/TEST10K_FLT_LGR_NNC"
|
|
case_path = case_root_path + "/TEST10K_FLT_LGR_NNC.EGRID"
|
|
case = rips_instance.project.load_case(path=case_path)
|
|
|
|
# Get the first view or create one
|
|
views = case.views()
|
|
if views:
|
|
view = views[0]
|
|
else:
|
|
view = case.create_view()
|
|
|
|
# Test visible_cells at time step 0
|
|
visibility = view.visible_cells(time_step=0)
|
|
|
|
# Verify we got a list
|
|
assert isinstance(visibility, list)
|
|
|
|
# Verify we have data
|
|
assert len(visibility) > 0
|
|
|
|
# Verify all values are either 0 or 1
|
|
for v in visibility:
|
|
assert v in [0, 1]
|
|
|
|
# Count visible and invisible cells
|
|
visible_count = sum(visibility)
|
|
invisible_count = len(visibility) - visible_count
|
|
|
|
# Total cells should match grid size
|
|
assert len(visibility) == visible_count + invisible_count
|
|
|
|
# The visibility array should have the correct size (can be all visible, all invisible, or mixed)
|
|
# This depends on the view settings
|
|
assert visible_count + invisible_count > 0
|
|
|
|
|
|
def test_visible_cells_different_timesteps(rips_instance, initialize_test):
|
|
"""Test visible_cells() at different time steps"""
|
|
case_root_path = dataroot.PATH + "/TEST10K_FLT_LGR_NNC"
|
|
case_path = case_root_path + "/TEST10K_FLT_LGR_NNC.EGRID"
|
|
case = rips_instance.project.load_case(path=case_path)
|
|
|
|
view = case.views()[0] if case.views() else case.create_view()
|
|
|
|
# Test at time step 0
|
|
visibility_ts0 = view.visible_cells(time_step=0)
|
|
assert len(visibility_ts0) > 0
|
|
|
|
# Test at time step 1 (if it exists)
|
|
visibility_ts1 = view.visible_cells(time_step=1)
|
|
assert len(visibility_ts1) > 0
|
|
|
|
# Both should have the same number of cells
|
|
assert len(visibility_ts0) == len(visibility_ts1)
|
|
|
|
|
|
def test_export_contour_map_to_text_requires_contour_map(
|
|
rips_instance, initialize_test
|
|
):
|
|
case_path = dataroot.PATH + "/TEST10K_FLT_LGR_NNC/TEST10K_FLT_LGR_NNC.EGRID"
|
|
case = rips_instance.project.load_case(case_path)
|
|
view = case.create_view()
|
|
with tempfile.TemporaryDirectory(prefix="rips") as tmpdirname:
|
|
# A regular 3D view is not a contour map
|
|
with pytest.raises(rips.RipsError):
|
|
view.export_contour_map_to_text(
|
|
export_file_name=os.path.join(tmpdirname, "map.txt")
|
|
)
|
|
with pytest.raises(rips.RipsError):
|
|
view.export_contour_map_to_text()
|