diff options
| -rw-r--r-- | test/stress_test.py | 6 |
1 files changed, 4 insertions, 2 deletions
diff --git a/test/stress_test.py b/test/stress_test.py index b339f36..749fdf3 100644 --- a/test/stress_test.py +++ b/test/stress_test.py @@ -5,6 +5,7 @@ import math from multiprocessing import Pool import random import subprocess +from subprocess import PIPE import sys from timeit import default_timer @@ -35,7 +36,8 @@ def timer(description): def run_client(i, client, stdin): with timer(f"Invocation #{i}"): cmd = client.get_command_line() - result = subprocess.run(cmd, text=True, input=stdin, capture_output=True) + result = subprocess.run(cmd, stdout=PIPE, stderr=PIPE, input=stdin, + universal_newlines=True) result.check_returncode() return result @@ -72,7 +74,7 @@ def run_stress_test(args): results = pool.starmap(run_client, [(i, client, stdin) for i in range(args.processes)]) assert results # It's assumed every invocation gives the same output, yikes. - actual_output = list(map(float, results[0].stdout.split('\n\n')[:-1])) + actual_output = list(map(float, results[0].stdout.split('\n')[:-1])) assert len(expected_output) == len(actual_output) for i in range(len(expected_output)): if math.isclose(expected_output[i], actual_output[i]): |