add tests for regex

Author: JavierCladellas

tests/scalability/test_scalabilityHandler.py

@@ -2,7 +2,7 @@
 
 import pytest
 import tempfile, json
-from feelpp.benchmarking.reframe.scalability import ScalabilityHandler, CsvExtractor,TsvExtractor,JsonExtractor,Extractor,ExtractorFactory
+from feelpp.benchmarking.reframe.scalability import ScalabilityHandler, CsvExtractor,TsvExtractor,JsonExtractor,RegexExtractor,Extractor,ExtractorFactory
 import numpy as np
 
 class StageMocker:
@@ -83,6 +83,33 @@ def test_extractTsv(self):
         file.close()
 
 
+
+    def test_extractRegex(self):
+        """ Test extracting performance variables using regex from a file """
+
+        file = tempfile.NamedTemporaryFile(mode="w+")
+        content = "assembly: 0.012\nsolve: 1.42\npostprocess: 0.08"
+        file.write(content)
+        file.flush()
+
+        pattern = r"^(?P<name>[^:]+):\s*(?P<value>[\d.]+)$"
+        extractor = RegexExtractor(
+            filepath=file.name,
+            stage_name="timers",
+            pattern=pattern,
+            variable_name_group="name",
+            variable_value_group="value",
+            units={"*":"s"}
+        )
+        perfvars = extractor.extract()
+        assert perfvars["timers_assembly"].evaluate() == 0.012
+        assert perfvars["timers_solve"].evaluate() == 1.42
+        assert perfvars["timers_postprocess"].evaluate() == 0.08
+
+        file.close()
+
+
+
     def test_extractJson(self):
         """ Test performance variable extraction for JSON files"""
         file = tempfile.NamedTemporaryFile()

tests/scalability/test_scalabilityValidation.py

@@ -25,5 +25,29 @@ def test_format(self):
         stage = Stage(**{"name":"test_stage","filepath":"test_filepath","format":"csv"})
         assert stage.variables_path == []
 
+    def test_regex_validation(self):
+        """ Tests mandatory regex fields and named/numbered groups """
+        # Missing pattern
+        with pytest.raises(ValidationError, match="regex must be specified if format == regex"):
+            Stage(**{"name": "r_stage", "filepath": "file.txt", "format": "regex", "variable_value_group": "value"})
+
+        # Missing variable_value_group
+        with pytest.raises(ValidationError, match="variable_value_group must be specified if format == regex"):
+            Stage(**{"name": "r_stage", "filepath": "file.txt", "format": "regex", "pattern": ".*"})
+
+        # Valid named capture groups
+        stage = Stage(
+            name="r_stage",
+            filepath="file.txt",
+            format="regex",
+            pattern="^(?P<name>[^:]+):\\s*(?P<value>[\\d.]+)$",
+            variable_name_group="name",
+            variable_value_group="value"
+        )
+        assert stage.format == "regex"
+        assert stage.variable_name_group == "name"
+        assert stage.variable_value_group == "value"
+
+
 class TestAppOutput:
     pass