WellDrillingCostCorrelation enum equality

Author: softwareengineerprogrammer

src/geophires_x/OptionList.py

@@ -110,6 +110,9 @@ def __init__(self, idx: int, _: str, c2:float, c1:float, c0:float):
         self._c1 = c1
         self._c0 = c0
 
+    def __eq__(self, other):
+        return str(self) == str(other)
+
     def calculate_cost_MUSD(self, meters) -> float:
         return (self._c2 * meters ** 2 + self._c1 * meters + self._c0) * 1E-6
 

tests/geophires_x_tests/test_options_list.py

@@ -0,0 +1,32 @@
+from base_test_case import BaseTestCase
+from geophires_x.OptionList import EndUseOptions
+from geophires_x.OptionList import PlantType
+from geophires_x.OptionList import WellDrillingCostCorrelation
+
+
+class EndUseOptionsTestCase(BaseTestCase):
+    def test_get_end_use_option_from_input_string(self):
+        self.assertEqual(EndUseOptions.get_end_use_option_from_input_string('1'), EndUseOptions.ELECTRICITY)
+
+        with self.assertRaises(ValueError):
+            EndUseOptions.get_end_use_option_from_input_string('2034982309')
+
+    def test_equality(self):
+        self.assertFalse(EndUseOptions.HEAT == EndUseOptions.ELECTRICITY)
+        self.assertTrue(EndUseOptions.HEAT == EndUseOptions.HEAT)
+        self.assertFalse(EndUseOptions.HEAT is None)
+        self.assertTrue(EndUseOptions.HEAT is EndUseOptions.HEAT)
+        # self.assertTrue(EndUseOptions.HEAT == 'HEAT')
+        # self.assertFalse(EndUseOptions.HEAT == 'Electricity')
+
+
+class WellDrillingCostCorrelationTestCase(BaseTestCase):
+    def test_equality(self):
+        self.assertFalse(WellDrillingCostCorrelation.VERTICAL_SMALL == WellDrillingCostCorrelation.DEVIATED_SMALL)
+        self.assertTrue(WellDrillingCostCorrelation.VERTICAL_SMALL == WellDrillingCostCorrelation.VERTICAL_SMALL)
+
+
+class PlantTypeTestCase(BaseTestCase):
+    def test_equality(self):
+        self.assertFalse(PlantType.SUB_CRITICAL_ORC == PlantType.SUPER_CRITICAL_ORC)
+        self.assertTrue(WellDrillingCostCorrelation.VERTICAL_SMALL == WellDrillingCostCorrelation.VERTICAL_SMALL)