From 34cd9c26126231eeb25f93af245b0a24e8d45ed8 Mon Sep 17 00:00:00 2001 From: Jason Date: Wed, 15 Jul 2020 02:07:15 -0400 Subject: [PATCH] Match parse() return tuple order to input tuple order (#89) Fixes #88. --- src/wireviz/wireviz.py | 13 +++++++------ 1 file changed, 7 insertions(+), 6 deletions(-) diff --git a/src/wireviz/wireviz.py b/src/wireviz/wireviz.py index b410c65..d52d3b1 100755 --- a/src/wireviz/wireviz.py +++ b/src/wireviz/wireviz.py @@ -185,12 +185,13 @@ def parse(yaml_input, file_out=None, return_types: (None, str, Tuple[str]) = Non return_types = [t.lower() for t in return_types] - if 'png' in return_types: - returns.append(harness.png) - if 'svg' in return_types: - returns.append(harness.svg) - if 'harness' in return_types: - returns.append(harness) + for rt in return_types: + if rt == 'png': + returns.append(harness.png) + if rt == 'svg': + returns.append(harness.svg) + if rt == 'harness': + returns.append(harness) return tuple(returns) if len(returns) != 1 else returns[0]