From 0d8ffed538d5f501b2328d58fa07e88c5d2ca9bb Mon Sep 17 00:00:00 2001 From: Egor Tensin Date: Fri, 13 Dec 2019 07:00:28 +0300 Subject: stress_test.py: fix for 3.6 --- test/stress_test.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) (limited to 'test') 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]): -- cgit v1.2.3