extend contingency test to cover heat pump & absorption chiller

Author: softwareengineerprogrammer

tests/test_geophires_x.py

@@ -1121,41 +1121,63 @@ def capex(result_cap_costs):
             return result_cap_costs['Total capital costs']['value']
 
         default_contingency_percent = 15
-        result_default = _get_result()
-
-        self.assertEqual(
-            # Test assumption check, update default_contingency_percent if GEOPHIRES default value is changed
-            capex(result_default),
-            capex(_get_result(contingency_percentage=default_contingency_percent)),
-        )
 
         for higher_contingency in range(20, 35, 5):
             assert higher_contingency > default_contingency_percent  # test assumption check
-            result_higher_contingency = _get_result(contingency_percentage=higher_contingency)
 
-            self.assertGreater(
-                capex(result_higher_contingency),
-                capex(result_default),
-            )
-            self.assertEqual(
-                # Contingency is not applied to drilling costs
-                result_default['Drilling and completion costs']['value'],
-                result_higher_contingency['Drilling and completion costs']['value'],
-            )
-
-            default_contingency_factor = 1.0 - (default_contingency_percent / 100.0)
-            higher_contingency_factor = 1 + (higher_contingency / 100.0)
-            for cost_category in [
-                'Stimulation costs',
-                'Surface power plant costs',
-                'Field gathering system costs',
-                'Total surface equipment costs',
-                'Exploration costs',
+            for input_file_path_ in [
+                'geophires_x_tests/generic-egs-case.txt',
+                'examples/example10_HP.txt',
+                'examples/example11_AC.txt',
             ]:
-                self.assertAlmostEqualWithinPercentage(
-                    result_default[cost_category]['value'] * default_contingency_factor * higher_contingency_factor,
-                    result_higher_contingency[cost_category]['value'],
-                    percent=min(  # Rounding throws off by a few percent
-                        2.5, (higher_contingency - default_contingency_percent) / 2.0
-                    ),
-                )
+                with self.subTest(msg=f'higher_contingency={higher_contingency}, input_file_path={input_file_path_}'):
+                    result_default = _get_result(input_file_path=input_file_path_)
+
+                    self.assertEqual(
+                        # Test assumption check, update default_contingency_percent
+                        # if GEOPHIRES default value is changed.
+                        capex(result_default),
+                        capex(
+                            _get_result(
+                                contingency_percentage=default_contingency_percent,
+                                input_file_path=input_file_path_,
+                            )
+                        ),
+                    )
+
+                    result_higher_contingency = _get_result(
+                        contingency_percentage=higher_contingency, input_file_path=input_file_path_
+                    )
+
+                    self.assertGreater(
+                        capex(result_higher_contingency),
+                        capex(result_default),
+                    )
+                    self.assertEqual(
+                        # Contingency is not applied to drilling costs
+                        result_default['Drilling and completion costs']['value'],
+                        result_higher_contingency['Drilling and completion costs']['value'],
+                    )
+
+                    default_contingency_factor = 1.0 - (default_contingency_percent / 100.0)
+                    higher_contingency_factor = 1 + (higher_contingency / 100.0)
+                    for cost_category in [
+                        'Stimulation costs',
+                        'Surface power plant costs',
+                        'Field gathering system costs',
+                        'Total surface equipment costs',
+                        'Exploration costs',
+                    ]:
+                        expected = (
+                            result_default[cost_category]['value']
+                            * default_contingency_factor
+                            * higher_contingency_factor
+                        )
+
+                        self.assertAlmostEqualWithinPercentage(
+                            expected,
+                            result_higher_contingency[cost_category]['value'],
+                            percent=min(  # Rounding throws off by a few percent
+                                4.5, (higher_contingency - default_contingency_percent) / 2.0
+                            ),
+                        )