update

Author: zrr1999

tools/get_cases_from_csv.py

@@ -4,7 +4,8 @@
 
 app = typer.Typer()
 
-def _get_cases(api_name: str, original_csv: str):
+
+def _get_cases(api_name: str, only_diff: bool, original_csv: str):
     with (
         open(original_csv, "r") as infile,
         open(f"filtered_result_{api_name}.csv", "w", newline="") as outfile,
@@ -26,7 +27,7 @@ def _get_cases(api_name: str, original_csv: str):
             last_col = float(row[-1]) if row[-1].strip() else 0
             second_last_col = float(row[-2]) if row[-2].strip() else 0
 
-            if last_col < 1e-16 and second_last_col < 1e-16:
+            if only_diff and last_col < 1e-16 and second_last_col < 1e-16:
                 continue
 
             writer.writerow(row)
@@ -44,22 +45,19 @@ def _get_cases(api_name: str, original_csv: str):
             first_col = row[0]
             last_col = float(row[-1]) if row[-1].strip() else 0
             second_last_col = float(row[-2]) if row[-2].strip() else 0
-
-            if last_col < 1e-16 and second_last_col < 1e-16:
-                continue
             outs.append(row[2].replace('""', '"'))
         outfile.write("\n".join(outs))
 
 
 @app.command()
 def get_cases(
     api_names: list[str],
+    only_diff: bool = True,
     config_path: Path = Path("TotalStableFull.csv"),
 ):
     for api_name in api_names:
-        _get_cases(api_name, config_path.as_posix())
-
+        _get_cases(api_name, only_diff, config_path.as_posix())
 
 
 if __name__ == "__main__":
-    app()
+    app()