read Number of Doublets as Well Bores parameter (WIP)

Author: softwareengineerprogrammer

src/geophires_x/WellBores.py

@@ -713,7 +713,7 @@ def __init__(self, model: Model):
             DefaultValue=2,
             AllowableRange=list(range(1, 201, 1)),
             UnitType=Units.NONE,
-            Required=True,
+            Required=False,
             ErrMessage="assume default number of production wells (2)",
             ToolTipText="Number of (identical) production wells"
         )
@@ -722,10 +722,17 @@ def __init__(self, model: Model):
             DefaultValue=2,
             AllowableRange=list(range(0, 201, 1)),
             UnitType=Units.NONE,
-            Required=True,
+            Required=False,
             ErrMessage="assume default number of injection wells (2)",
             ToolTipText="Number of (identical) injection wells"
         )
+        self.doublets_count = self.ParameterDict[self.doublets_count.Name] = intParameter(
+            "Number of Doublets",
+            DefaultValue=2,
+            AllowableRange=list(range(0, 201, 1)),
+            UnitType=Units.NONE,
+            ToolTipText="Number of doublets"  # FIXME WIP
+        )
         self.prodwelldiam = self.ParameterDict[self.prodwelldiam.Name] = floatParameter(
             "Production Well Diameter",
             DefaultValue=8.0,
@@ -1317,6 +1324,11 @@ def read_parameters(self, model: Model) -> None:
 
         coerce_int_params_to_enum_values(self.ParameterDict)
 
+        if self.doublets_count.Provided:
+            # FIXME WIP validate that ninj/nprod haven't been provided and vice versa
+            self.ninj.value = self.doublets_count.value
+            self.nprod.value = self.doublets_count.value
+
         model.logger.info(f"read parameters complete {self.__class__.__name__}: {__name__}")
 
     def Calculate(self, model: Model) -> None:

tests/geophires_x_tests/test_well_bores.py

@@ -0,0 +1,51 @@
+from __future__ import annotations
+
+
+from base_test_case import BaseTestCase
+
+# ruff: noqa: I001  # Successful module initialization is dependent on this specific import order.
+
+# noinspection PyProtectedMember
+
+# noinspection PyProtectedMember
+from geophires_x_client import GeophiresInputParameters
+from geophires_x_client import GeophiresXClient
+from geophires_x_client import GeophiresXResult
+
+
+class WellBoresTestCase(BaseTestCase):
+
+    def test_number_of_doublets(self):
+        def _get_result(_params) -> GeophiresXResult:
+            params = GeophiresInputParameters(
+                {
+                    'Reservoir Depth': 5,
+                    'Gradient 1': 74,
+                    'Power Plant Type': 2,
+                    'Maximum Temperature': 600,
+                }
+                | _params
+            )
+            return GeophiresXClient().get_geophires_result(params)
+
+        def _prod_inj(_r: GeophiresXResult) -> tuple[int, int]:
+            return (
+                _r.result['ENGINEERING PARAMETERS']['Number of Production Wells']['value'],
+                _r.result['ENGINEERING PARAMETERS']['Number of Injection Wells']['value'],
+            )
+
+        r_prod_inj: GeophiresXResult = _get_result(
+            {
+                'Number of Production Wells': 10,
+                'Number of Injection Wells': 10,
+            }
+        )
+
+        r_doublets: GeophiresXResult = _get_result(
+            {
+                'Number of Doublets': 10,
+            }
+        )
+
+        self.assertEqual(_prod_inj(r_doublets), _prod_inj(r_prod_inj))
+        # FIXME WIP assert results look superficially equal