From b924e3d8f867c6527983d8ea0af1f0cbadbada3f Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Pawe=C5=82=20Bylica?= Date: Fri, 26 Apr 2024 17:46:02 +0200 Subject: [PATCH] improve code generation --- tests/prague/eip663_dupn_swapn_exchange/test_dupn.py | 4 ++-- tests/prague/eip663_dupn_swapn_exchange/test_swapn.py | 6 +++--- 2 files changed, 5 insertions(+), 5 deletions(-) diff --git a/tests/prague/eip663_dupn_swapn_exchange/test_dupn.py b/tests/prague/eip663_dupn_swapn_exchange/test_dupn.py index f928211b78..6efdd493c9 100644 --- a/tests/prague/eip663_dupn_swapn_exchange/test_dupn.py +++ b/tests/prague/eip663_dupn_swapn_exchange/test_dupn.py @@ -32,11 +32,11 @@ def test_dupn_all_valid_immediates( sections=[ Section.Code( code=b"".join(Op.PUSH2(v) for v in values) - + b"".join(Op.DUPN(x) + Op.PUSH2(x) + Op.SSTORE for x in range(0, n)) + + b"".join(Op.SSTORE(x, Op.DUPN[x]) for x in range(0, n)) + Op.STOP, code_inputs=0, code_outputs=NON_RETURNING_SECTION, - max_stack_height=258, + max_stack_height=n + 2, ) ], ) diff --git a/tests/prague/eip663_dupn_swapn_exchange/test_swapn.py b/tests/prague/eip663_dupn_swapn_exchange/test_swapn.py index 81a5b5b671..9975578978 100644 --- a/tests/prague/eip663_dupn_swapn_exchange/test_swapn.py +++ b/tests/prague/eip663_dupn_swapn_exchange/test_swapn.py @@ -25,18 +25,18 @@ def test_swapn_all_valid_immediates( """ Test case for all valid SWAPN immediates. """ - n = 100 + n = 256 values = range(0x500, 0x500 + 257) eof_code = Container( sections=[ Section.Code( code=b"".join(Op.PUSH2(v) for v in values) - + b"".join(Op.SWAPN(0xFF - x) + Op.PUSH2(x) + Op.SSTORE for x in range(0, n)) + + b"".join(Op.SSTORE(x, Op.SWAPN[0xFF - x]) for x in range(0, n)) + Op.STOP, code_inputs=0, code_outputs=NON_RETURNING_SECTION, - max_stack_height=258, + max_stack_height=n + 2, ) ], )