diff options
Diffstat (limited to 'Tools/cases_generator/stack.py')
| -rw-r--r-- | Tools/cases_generator/stack.py | 11 |
1 files changed, 6 insertions, 5 deletions
diff --git a/Tools/cases_generator/stack.py b/Tools/cases_generator/stack.py index c0e1278..ebe62df 100644 --- a/Tools/cases_generator/stack.py +++ b/Tools/cases_generator/stack.py @@ -28,14 +28,15 @@ def var_size(var: StackItem) -> str: if var.condition == "0": return "0" elif var.condition == "1": - return var.size - elif var.condition == "oparg & 1" and var.size == "1": + return var.get_size() + elif var.condition == "oparg & 1" and not var.size: return f"({var.condition})" else: - return f"(({var.condition}) ? {var.size} : 0)" - else: + return f"(({var.condition}) ? {var.get_size()} : 0)" + elif var.size: return var.size - + else: + return "1" @dataclass class StackOffset: |