Commit ff959f13 authored by Jan Kiene's avatar Jan Kiene
Browse files

add sorting to output of parse_xml_report.py

parent ec73f1bd
Loading
Loading
Loading
Loading
Loading
+9 −0
Original line number Diff line number Diff line
@@ -18,6 +18,10 @@ class Result(str, Enum):
    PASS = "PASS"


SORT_KEYS = ["MAX_ABS_DIFF", "MLD", "MIN_SSNR", "MIN_ODG"]
SORT_ASCENDING = [False, False, True, True]


class TestcaseParser(dict):
    def __init__(self, testcases: list):
        super().__init__()
@@ -111,6 +115,11 @@ class TestcaseParser(dict):
    def to_df(self) -> pd.DataFrame:
        testcases = list(self.values())
        df = pd.DataFrame(testcases)

        for key in SORT_KEYS:
            df[key] = pd.to_numeric(df[key], errors="coerce")
        df = df.sort_values(by=SORT_KEYS, ascending=SORT_ASCENDING)

        return df