From 348f91e607ab3b6f05968147691eda4cb7cff2be Mon Sep 17 00:00:00 2001 From: "Jason R. Jones" Date: Tue, 14 Jul 2020 18:59:20 -0400 Subject: [PATCH] Fix the order of the returned variables, see issue #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]