feat: add utility function get_cosmic_ray_report_filtered() to filter cosmic-ray reports from KILLED mutant tests

aip_trainer/utils/split_cosmic_ray_report.py

@@ -0,0 +1,29 @@
+from pathlib import Path
+
+
+def get_cosmic_ray_report_filtered(input_filename, suffix="filtered", separator="============", filter_string="test outcome: TestOutcome.KILLED"):
+    filename, ext = Path(input_filename).stem, Path(input_filename).suffix
+    working_dir = input_filename.parent
+    # Read the input file
+    with open(input_filename, 'r') as file:
+        content = file.read()
+
+    # Split the content into sections
+    sections = content.split(separator)
+
+    # Filter out sections containing "test outcome: TestOutcome.KILLED"
+    filtered_sections = [section for section in sections if filter_string not in section]
+
+    # Join the filtered sections back into a single string
+    filtered_content = separator.join(filtered_sections)
+
+    # Write the filtered content to a new file
+    with open(working_dir / f'{filename}_{suffix}{ext}', 'w') as file:
+        file.write(filtered_content)
+
+
+
+if __name__ == "__main__":
+    from aip_trainer import PROJECT_ROOT_FOLDER
+    _input_filename =  "cosmic-ray-lambdagetsample4.txt"
+    get_cosmic_ray_report_filtered(PROJECT_ROOT_FOLDER / "tmp" / _input_filename)

tests/utils/test_split_cosmic_ray_report.py

@@ -0,0 +1,26 @@
+import unittest
+from pathlib import Path
+from aip_trainer.utils.split_cosmic_ray_report import get_cosmic_ray_report_filtered
+from aip_trainer.utils import utilities
+from tests import EVENTS_FOLDER
+
+
+class TestSplitCosmicRayReport(unittest.TestCase):
+    def test_get_cosmic_ray_report_filtered(self):
+        input_filename = EVENTS_FOLDER / "cosmic-ray-lambdagetsample.txt"
+        output_filename = EVENTS_FOLDER / f"{input_filename.stem}_filtered{input_filename.suffix}"
+        self.assertFalse(output_filename.exists())
+        get_cosmic_ray_report_filtered(input_filename, separator="============", filter_string="test outcome: TestOutcome.KILLED")
+
+        # Check if the filtered file is created
+        self.assertTrue(output_filename.exists() and output_filename.is_file())
+
+        # Verify the filtered content
+        hash_output = utilities.hash_calculate(output_filename, True)
+        self.assertEqual(hash_output, b'HqzO5Hq50Z1CIz/8rGE8Hpt4wSRM5Xm86pv+9FOYbUg=')
+
+        output_filename.unlink(missing_ok=False)
+
+
+if __name__ == "__main__":
+    unittest.main()