fix category row label case

Author: softwareengineerprogrammer

src/geophires_x/EconomicsSamCashFlow.py

@@ -116,7 +116,7 @@ def is_only_commas(s: str) -> bool:
                 # mapped.
                 continue
 
-            if re.match(r'^([A-Z \(\)\-\:]+)$', row_label) or re.match(r'^([A-Za-z \-]+\:)$', row_label):
+            if _is_category_row_label(row_label):
                 category_row(row_label)
                 continue
 
@@ -131,6 +131,10 @@ def is_only_commas(s: str) -> bool:
     return _clean_profile(profile)
 
 
+def _is_category_row_label(row_label: str) -> bool:
+    return re.match(r'^([A-Z \(\)\-\:]+)$', row_label) or re.match(r'^([A-Z][A-Za-z \-]+\:)$', row_label)
+
+
 def _clean_profile(profile: list[list[Any]]) -> list[list[Any]]:
     # Collapse consecutive blank rows
     previous_line_was_blank = False

tests/geophires_x_tests/test_economics_sam.py

@@ -21,7 +21,7 @@
 )
 
 # noinspection PyProtectedMember
-from geophires_x.EconomicsSamCashFlow import _clean_profile
+from geophires_x.EconomicsSamCashFlow import _clean_profile, _is_category_row_label
 from geophires_x_client import GeophiresInputParameters
 from geophires_x_client import GeophiresXClient
 from geophires_x_client import GeophiresXResult
@@ -162,6 +162,10 @@ def test_clean_profile(self):
             ],
         )
 
+    def test_is_category_row_label(self):
+        self.assertTrue(_is_category_row_label('OPERATING ACTIVITIES'))
+        self.assertFalse(_is_category_row_label('plus PBI if not available for debt service:'))
+
     def test_sig_figs(self):
         self.assertListEqual(_sig_figs([1.14, 2.24], 2), [1.1, 2.2])
         self.assertListEqual(_sig_figs((1.14, 2.24), 2), [1.1, 2.2])