diff --git a/run-tests.py b/run-tests.py index fb2f1e9..0f55361 100644 --- a/run-tests.py +++ b/run-tests.py @@ -26,7 +26,7 @@ for file_name in glob.glob("*.c", recursive=False): to_run = f.read().split("\n\n") total_tests, passed_tests = 0, 0 for x in to_run: - x_split = x.split("\n") + x_split = x.rsplit("\n", 1) return_data = subprocess.run(["./a.out"], input=x_split[0].rstrip().encode(), capture_output=True).stdout.decode().rstrip() test_passed_int = int(return_data == x_split[1].rstrip()) total_tests += 1