aboutsummaryrefslogtreecommitdiffstats
path: root/host/utils/converter_benchmark.py
diff options
context:
space:
mode:
authorMartin Braun <martin.braun@ettus.com>2020-05-04 10:50:02 -0700
committerAaron Rossetto <aaron.rossetto@ni.com>2020-05-12 12:04:11 -0500
commitb9576f9601ffa31ffe38ee44421289a9ce2db575 (patch)
tree6ce9ae03b925269075f23cd3e55233bad9d764cf /host/utils/converter_benchmark.py
parent38a7bf50a9c7abbdeb7d445baa8c0ed19f424ff7 (diff)
downloaduhd-b9576f9601ffa31ffe38ee44421289a9ce2db575.tar.gz
uhd-b9576f9601ffa31ffe38ee44421289a9ce2db575.tar.bz2
uhd-b9576f9601ffa31ffe38ee44421289a9ce2db575.zip
utils: Remove 'six' dependency, fix minor PyLint issues
This removes the 'six' dependency from converter_benchmark, thus making the utils subdirectory no longer depend on six. A few Python2-isms and PyLint issues were also fixed.
Diffstat (limited to 'host/utils/converter_benchmark.py')
-rw-r--r--host/utils/converter_benchmark.py17
1 files changed, 8 insertions, 9 deletions
diff --git a/host/utils/converter_benchmark.py b/host/utils/converter_benchmark.py
index 0f7f32937..d34b019b6 100644
--- a/host/utils/converter_benchmark.py
+++ b/host/utils/converter_benchmark.py
@@ -9,10 +9,10 @@
Wrap the converter_benchmark tool and produce prettier results.
"""
+import sys
import argparse
import csv
import subprocess
-from six import iteritems
INTRO_SETUP = {
'n_samples': {
@@ -38,7 +38,7 @@ TABLE_SETUP = {
def run_benchmark(args):
""" Run the tool with the given arguments, return the section in the {{{ }}} brackets """
call_args = ['./converter_benchmark',]
- for k, v in iteritems(args.__dict__):
+ for k, v in args.__dict__.items():
k = k.replace('_', '-')
if v is None:
continue
@@ -48,18 +48,17 @@ def run_benchmark(args):
continue
call_args.append('--{0}'.format(k))
call_args.append(str(v))
- print(call_args)
try:
output = subprocess.check_output(call_args).decode('utf-8')
except subprocess.CalledProcessError as ex:
- print(ex.output)
- exit(ex.returncode)
+ print("Error running converter_benchmark:", ex.output.decode())
+ sys.exit(ex.returncode)
header_out, csv_output = output.split('{{{', 1)
csv_output = csv_output.split('}}}', 1)
assert len(csv_output) == 2 and csv_output[1].strip() == ''
return header_out, csv_output[0]
-def print_stats_table(args, csv_output):
+def print_stats_table(_, csv_output):
"""
Print stats.
"""
@@ -69,7 +68,7 @@ def print_stats_table(args, csv_output):
for idx, row in enumerate(reader):
if idx == 0:
# Print intro:
- for k, v in iteritems(INTRO_SETUP):
+ for k, v in INTRO_SETUP.items():
print("{title}: {value}".format(
title=v['title'],
value=row[title_row.index(k)],
@@ -159,7 +158,8 @@ def setup_argparse():
)
parser.add_argument(
"--debug-converter", action='store_true',
- help="Skip benchmark and print conversion results. Implies iterations==1 and will only run on a single converter.",
+ help="Skip benchmark and print conversion results. Implies iterations==1 "
+ "and will only run on a single converter.",
)
parser.add_argument(
"--hex", action='store_true',
@@ -180,4 +180,3 @@ def main():
if __name__ == "__main__":
main()
-