mirror of
https://github.com/OPM/ResInsight.git
synced 2025-02-25 18:55:39 -06:00
116 lines
3.8 KiB
Python
116 lines
3.8 KiB
Python
import sys
|
|
import os
|
|
import math
|
|
import contextlib
|
|
import os
|
|
import shutil
|
|
import tempfile
|
|
|
|
sys.path.insert(1, os.path.join(sys.path[0], '../../'))
|
|
import rips
|
|
|
|
import dataroot
|
|
|
|
|
|
def test_summary_import_and_find(rips_instance, initialize_test):
|
|
casePath = dataroot.PATH + "/flow_diagnostics_test/SIMPLE_SUMMARY2.SMSPEC"
|
|
summary_case = rips_instance.project.import_summary_case(casePath)
|
|
assert(summary_case.id == 1)
|
|
|
|
case_id = 234
|
|
found_summary_case = rips_instance.project.summary_case(case_id)
|
|
assert(found_summary_case is None)
|
|
|
|
correct_case_id = 1
|
|
found_summary_case = rips_instance.project.summary_case(correct_case_id)
|
|
assert(found_summary_case is not None)
|
|
|
|
rips_instance.project.close()
|
|
correct_case_id = 1
|
|
found_summary_case = rips_instance.project.summary_case(correct_case_id)
|
|
assert(found_summary_case is None)
|
|
|
|
|
|
def test_summary_data(rips_instance, initialize_test):
|
|
casePath = dataroot.PATH + "/flow_diagnostics_test/SIMPLE_SUMMARY2.SMSPEC"
|
|
summary_case = rips_instance.project.import_summary_case(casePath)
|
|
assert(summary_case.id == 1)
|
|
|
|
addresses = summary_case.available_addresses()
|
|
assert(len(addresses.values) == 343)
|
|
|
|
summary_data = summary_case.summary_vector_values("FOPT")
|
|
assert(len(summary_data.values) == 60)
|
|
|
|
|
|
def test_summary_resample(rips_instance, initialize_test):
|
|
casePath = dataroot.PATH + "/flow_diagnostics_test/SIMPLE_SUMMARY2.SMSPEC"
|
|
summary_case = rips_instance.project.import_summary_case(casePath)
|
|
assert(summary_case.id == 1)
|
|
|
|
summary_data_sampled = summary_case.resample_values("FOPT", "NONE")
|
|
assert(len(summary_data_sampled.values) == 60)
|
|
assert(len(summary_data_sampled.time_steps) == 60)
|
|
|
|
summary_data_sampled = summary_case.resample_values("FOPT", "DAY")
|
|
assert(len(summary_data_sampled.values) == 721)
|
|
assert(len(summary_data_sampled.time_steps) == 721)
|
|
|
|
summary_data_sampled = summary_case.resample_values("FOPT", "MONTH")
|
|
assert(len(summary_data_sampled.values) == 24)
|
|
assert(len(summary_data_sampled.time_steps) == 24)
|
|
|
|
summary_data_sampled = summary_case.resample_values("FOPT", "QUARTER")
|
|
assert(len(summary_data_sampled.values) == 8)
|
|
assert(len(summary_data_sampled.time_steps) == 8)
|
|
|
|
summary_data_sampled = summary_case.resample_values("FOPT", "YEAR")
|
|
assert(len(summary_data_sampled.values) == 3)
|
|
assert(len(summary_data_sampled.time_steps) == 3)
|
|
|
|
|
|
@contextlib.contextmanager
|
|
def cd(newdir, cleanup=lambda: True):
|
|
prevdir = os.getcwd()
|
|
os.chdir(os.path.expanduser(newdir))
|
|
try:
|
|
yield
|
|
finally:
|
|
os.chdir(prevdir)
|
|
cleanup()
|
|
|
|
@contextlib.contextmanager
|
|
def tempdir():
|
|
dirpath = tempfile.mkdtemp()
|
|
def cleanup():
|
|
shutil.rmtree(dirpath)
|
|
with cd(dirpath, cleanup):
|
|
yield dirpath
|
|
|
|
# This test ensures that missing unsmry file is handeled gracefully
|
|
def test_summary_no_unsmry(rips_instance, initialize_test):
|
|
casePathRelative = dataroot.PATH + "/flow_diagnostics_test/SIMPLE_SUMMARY2.SMSPEC"
|
|
|
|
# create an absolute path, as the helper functions used to create a temporary folder does not work
|
|
# with the relative (..\..\) part of the file path
|
|
casePath = os.path.abspath(casePathRelative)
|
|
|
|
with tempdir() as dirpath:
|
|
base_path = os.path.basename(casePath)
|
|
temp_path = os.path.join(dirpath, base_path)
|
|
shutil.copy2(casePath, temp_path)
|
|
|
|
summary_case = rips_instance.project.import_summary_case(temp_path)
|
|
|
|
values = summary_case.summary_vector_values()
|
|
assert(len(values.values) == 1)
|
|
|
|
time_steps = summary_case.available_time_steps()
|
|
assert(len(time_steps.values) == 1)
|
|
|
|
addresses = summary_case.available_addresses()
|
|
assert(len(addresses.values) == 1)
|
|
|
|
summary_case.resample_values()
|
|
|