From 8ba8465bb16b5ad12196655d297bd8a14931f572 Mon Sep 17 00:00:00 2001 From: The Tumultuous Unicorn Of Darkness Date: Wed, 30 Apr 2025 21:31:56 +0200 Subject: [PATCH] Tests: allow to fix compressed files --- tests/run_tests.py | 12 +++++++++++- 1 file changed, 11 insertions(+), 1 deletion(-) diff --git a/tests/run_tests.py b/tests/run_tests.py index e71bbb0..526e97e 100755 --- a/tests/run_tests.py +++ b/tests/run_tests.py @@ -38,7 +38,17 @@ def fmt_error(err): return "{} expected `{}'\n{} got `{}'".format(pfix, err[1], ' '*len(pfix), err[2]) def fixFile(filename, input_lines, output_lines): - f = open(filename, "wt") + pfilename = Path(filename) + if len(pfilename.suffixes) >= 2: + # Compressed file + if pfilename.suffixes[1] == ".xz": + f = lzma.open(pfilename, "wt") + else: + print(f"Cannot fix {pfilename.name} because {''.join(pfilename.suffixes[1:])} is not supported") + return + else: + # Plain text file + f = open(pfilename, "wt") f.writelines([s + "\n" for s in input_lines]) f.write(delimiter + "\n") f.writelines([s + "\n" for s in output_lines])