diff --git a/benchmarks/utils.py b/benchmarks/utils.py index 932583d14aa33c1790ba2ce0b0acbc326f2bd633_YmVuY2htYXJrcy91dGlscy5weQ==..26e13d59a444de23e4a139b0f52f1bf450a2d8e6_YmVuY2htYXJrcy91dGlscy5weQ== 100644 --- a/benchmarks/utils.py +++ b/benchmarks/utils.py @@ -208,5 +208,5 @@ times.append(func(*args, **kwargs)) return median(times) - def _execute(self, command, check_returncode=True, *args): + def _execute(self, command, *args): cmd = self._prepare_cmd(command, *args) @@ -212,5 +212,5 @@ cmd = self._prepare_cmd(command, *args) - return self._timeit(self._single_execute, (cmd, check_returncode)) + return self._timeit(self._single_execute, (cmd,)) def _prepare_cmd(self, command, *args): cmd = [ @@ -221,5 +221,5 @@ return cmd - def _single_execute(self, cmd, check_returncode=True): + def _single_execute(self, cmd): before = time.time() @@ -225,17 +225,7 @@ before = time.time() - - if check_returncode is True: - try: - output = subprocess.check_output(cmd, stderr=subprocess.STDOUT, - env=self.environ) - after = time.time() - duration = after - before - return duration - except subprocess.CalledProcessError as e: - raise - else: - subprocess.call(cmd, stderr=subprocess.STDOUT, - env=self.environ) + try: + output = subprocess.check_output(cmd, stderr=subprocess.STDOUT, + env=self.environ) after = time.time() duration = after - before return duration @@ -239,8 +229,8 @@ after = time.time() duration = after - before return duration - - + except subprocess.CalledProcessError as e: + raise class BaseTimeTestSuite(BaseTestSuite):