From 906c9df10dbd3e7d4c28fe6fab2f3027c0429eb1 Mon Sep 17 00:00:00 2001 From: drunkendog Date: Tue, 12 Sep 2023 14:04:53 +0100 Subject: [PATCH] Update run-tests.py --- run-tests.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/run-tests.py b/run-tests.py index ec2028b..fb95690 100644 --- a/run-tests.py +++ b/run-tests.py @@ -23,5 +23,5 @@ for file_name in glob.glob("*.c", recursive=False): to_run = f.read().split("\n\n") for x in to_run: x_split = x.split("\n") - return_data = subprocess.run(["./a.out"], input=x[0].rstrip().encode(), capture_output=True).stdout.decode() - print(f"\u001b[{31 + int(return_data.rstrip() == x[1].rstrip())}m{return_data}\u001b[0m", x[1]) + return_data = subprocess.run(["./a.out"], input=x_split[0].rstrip().encode(), capture_output=True).stdout.decode().rstrip() + print(f"\u001b[{31 + int(return_data == x_split[1].rstrip())}m{return_data}\u001b[0m", x[1])