Skip to content

Commit 538076f

Browse files
Standardize from_input_string naming convention
1 parent 32fadc5 commit 538076f

File tree

4 files changed

+13
-13
lines changed

4 files changed

+13
-13
lines changed

src/geophires_x/OptionList.py

Lines changed: 8 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -31,14 +31,14 @@ class EndUseOptions(GeophiresInputEnum):
3131
COGENERATION_PARALLEL_EXTRA_ELECTRICITY = 52, "Cogeneration Parallel Cycle, Electricity sales considered as extra income"
3232

3333
@staticmethod
34-
def get_end_use_option_from_input_string(input_string:str):
34+
def from_input_string(input_string:str):
3535
"""
3636
:rtype: EndUseOptions
3737
"""
3838

39-
for option in EndUseOptions:
40-
if input_string == str(option.int_value):
41-
return option
39+
for member in __class__:
40+
if input_string == str(member.int_value):
41+
return member
4242

4343
raise ValueError(f'Unknown End-Use Option input value: {input_string}')
4444

@@ -61,14 +61,14 @@ class PlantType(GeophiresInputEnum):
6161
INDUSTRIAL = 9, "Industrial"
6262

6363
@staticmethod
64-
def get_plant_type_from_input_string(input_string:str):
64+
def from_input_string(input_string:str):
6565
"""
6666
:rtype: PlantType
6767
"""
6868

69-
for type in PlantType:
70-
if input_string == str(type.int_value):
71-
return type
69+
for member in __class__:
70+
if input_string == str(member.int_value):
71+
return member
7272

7373
raise ValueError(f'Unknown Power Plant Type input value: {input_string}')
7474

src/geophires_x/SurfacePlant.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -515,13 +515,13 @@ def read_parameters(self, model:Model) -> None:
515515

516516
# handle special cases
517517
if ParameterToModify.Name == 'End-Use Option':
518-
end_use_option = EndUseOptions.get_end_use_option_from_input_string(ParameterReadIn.sValue)
518+
end_use_option = EndUseOptions.from_input_string(ParameterReadIn.sValue)
519519
ParameterToModify.value = end_use_option
520520
if end_use_option == EndUseOptions.HEAT:
521521
self.plant_type.value = PlantType.INDUSTRIAL
522522

523523
elif ParameterToModify.Name == 'Power Plant Type':
524-
ParameterToModify.value = PlantType.get_plant_type_from_input_string(ParameterReadIn.sValue)
524+
ParameterToModify.value = PlantType.from_input_string(ParameterReadIn.sValue)
525525
if self.enduse_option.value == EndUseOptions.ELECTRICITY:
526526
# simple single- or double-flash power plant assumes no production well pumping
527527
if ParameterToModify.value in [PlantType.SINGLE_FLASH, PlantType.DOUBLE_FLASH]:

src/geophires_x/SurfacePlantAGS.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -298,7 +298,7 @@ def read_parameters(self, model: Model) -> None:
298298

299299
# handle special cases
300300
if ParameterToModify.Name == "End-Use Option":
301-
ParameterToModify.value = EndUseOptions.get_end_use_option_from_input_string(ParameterReadIn.sValue)
301+
ParameterToModify.value = EndUseOptions.from_input_string(ParameterReadIn.sValue)
302302

303303
else:
304304
model.logger.info("No parameters read because no content provided")

tests/geophires_x_tests/test_options_list.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -6,10 +6,10 @@
66

77
class EndUseOptionsTestCase(BaseTestCase):
88
def test_get_end_use_option_from_input_string(self):
9-
self.assertEqual(EndUseOptions.get_end_use_option_from_input_string('1'), EndUseOptions.ELECTRICITY)
9+
self.assertEqual(EndUseOptions.from_input_string('1'), EndUseOptions.ELECTRICITY)
1010

1111
with self.assertRaises(ValueError):
12-
EndUseOptions.get_end_use_option_from_input_string('2034982309')
12+
EndUseOptions.from_input_string('2034982309')
1313

1414
def test_get_end_use_option_from_int_val(self):
1515
self.assertEqual(EndUseOptions.from_int(1), EndUseOptions.ELECTRICITY)

0 commit comments

Comments
 (0)