Skip to content

Commit 23f35f2

Browse files
fix pressure/depth input/output categorization logic and add coverage for HipRaXResult
1 parent af31db7 commit 23f35f2

2 files changed

Lines changed: 14 additions & 12 deletions

File tree

src/hip_ra_x/hip_ra_x.py

Lines changed: 9 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -834,15 +834,14 @@ def render_scientific(p: Parameter) -> str:
834834

835835
inputs = self._get_output_config_for_summary_of_inputs_category(render_default, render_scientific)
836836
# If depth and/or pressure are provided, report them as inputs. If not, as outputs
837-
if self.reservoir_depth.Provided:
838-
# inputs.append((self.reservoir_depth, render_default))
839-
reservoir_pressure_idx = [it[0].Name for it in inputs].index(self.reservoir_pressure.Name)
840-
inputs.pop(reservoir_pressure_idx)
841-
if self.reservoir_pressure.Provided:
842-
# inputs.append((self.reservoir_pressure, render_default))
837+
if not self.reservoir_depth.Provided:
843838
reservoir_depth_idx = [it[0].Name for it in inputs].index(self.reservoir_depth.Name)
844839
inputs.pop(reservoir_depth_idx)
845840

841+
if not self.reservoir_pressure.Provided:
842+
reservoir_pressure_idx = [it[0].Name for it in inputs].index(self.reservoir_pressure.Name)
843+
inputs.pop(reservoir_pressure_idx)
844+
846845
for param, render in inputs:
847846
summary_of_inputs[param.Name] = render(param)
848847

@@ -851,14 +850,12 @@ def render_scientific(p: Parameter) -> str:
851850
outputs = self._get_output_config_for_summary_of_results_category(render_default, render_scientific)
852851

853852
# If depth and/or pressure are provided, report them as inputs. If not, as outputs
854-
if not self.reservoir_depth.Provided:
855-
# outputs.insert(0, (self.reservoir_depth, render_default))
856-
reservoir_pressure_idx = [it[0].Name for it in outputs].index(self.reservoir_pressure.Name)
857-
outputs.pop(reservoir_pressure_idx)
858-
if not self.reservoir_pressure.Provided:
859-
# outputs.insert(0, (self.reservoir_pressure, render_default))
853+
if self.reservoir_depth.Provided:
860854
reservoir_depth_idx = [it[0].Name for it in outputs].index(self.reservoir_depth.Name)
861855
outputs.pop(reservoir_depth_idx)
856+
if self.reservoir_pressure.Provided:
857+
reservoir_pressure_idx = [it[0].Name for it in outputs].index(self.reservoir_pressure.Name)
858+
outputs.pop(reservoir_pressure_idx)
862859

863860
for param, render in outputs:
864861
summary_of_results[param.Name] = render(param)

tests/hip_ra_x_tests/test_hip_ra_x.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -27,6 +27,7 @@
2727
from hip_ra import HipRaResult
2828
from hip_ra.HIP_RA import HIP_RA
2929
from hip_ra_x import HipRaXClient
30+
from hip_ra_x import HipRaXResult
3031
from hip_ra_x.hip_ra_x import HIP_RA_X
3132
from tests.base_test_case import BaseTestCase
3233

@@ -54,6 +55,10 @@ def get_output_file_for_example(example_file: Path):
5455
expected_result = HipRaResult(expected_result_output_file_path)
5556
self.assertDictEqual(expected_result.result, result.result)
5657

58+
x_result = client.get_hip_ra_x_result(HipRaInputParameters(input_file_path))
59+
expected_x_result = HipRaXResult.from_hip_ra_result(expected_result)
60+
self.assertDictEqual(expected_x_result.result, x_result.result)
61+
5762
# TODO
5863
# self.assertFileContentsEqual(expected_result_output_file_path, result.output_file_path)
5964

0 commit comments

Comments
 (0)