Fix the order of the returned variables, see issue #88

This commit is contained in:
Jason R. Jones 2020-07-14 18:59:20 -04:00
parent c8dc697845
commit 348f91e607

View File

@ -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] return_types = [t.lower() for t in return_types]
if 'png' in return_types: for rt in return_types:
returns.append(harness.png) if rt == 'png':
if 'svg' in return_types: returns.append(harness.png)
returns.append(harness.svg) if rt == 'svg':
if 'harness' in return_types: returns.append(harness.svg)
returns.append(harness) if rt == 'harness':
returns.append(harness)
return tuple(returns) if len(returns) != 1 else returns[0] return tuple(returns) if len(returns) != 1 else returns[0]