Skip to content

Commit 09f9ca4

Browse files
refactor: update worst jump test case
1 parent fe1efbe commit 09f9ca4

File tree

1 file changed

+3
-16
lines changed

1 file changed

+3
-16
lines changed

tests/zkevm/test_worst_compute.py

Lines changed: 3 additions & 16 deletions
Original file line numberDiff line numberDiff line change
@@ -27,7 +27,6 @@
2727
Transaction,
2828
add_kzg_version,
2929
)
30-
from ethereum_test_tools.code.generators import While
3130
from ethereum_test_tools.vm.opcode import Opcodes as Op
3231
from ethereum_test_types import TransactionType
3332
from ethereum_test_vm.opcode import Opcode
@@ -762,24 +761,12 @@ def code_loop_precompile_call(calldata: Bytecode, attack_block: Bytecode, fork:
762761
def test_worst_jumps(state_test: StateTestFiller, pre: Alloc, fork: Fork):
763762
"""Test running a JUMP-intensive contract."""
764763
env = Environment()
765-
max_code_size = fork.max_code_size()
766-
767-
def jump_seq(dest: int):
768-
return Op.JUMP(Op.PUSH2(dest)) + Op.JUMPDEST
769-
770-
bytes_per_seq = len(jump_seq())
771-
seqs_per_call = max_code_size // bytes_per_seq
772-
773-
# Create and deploy the jump-intensive contract
774-
jumps_code = sum([jump_seq(4 + 5 * i) for i in range(seqs_per_call)])
775-
jumps_address = pre.deploy_contract(code=bytes(jumps_code))
776764

777-
# Call the contract repeatedly until gas runs out.
778-
caller_code = While(body=Op.POP(Op.CALL(address=jumps_address)))
779-
caller_address = pre.deploy_contract(caller_code)
765+
jumps_code = Op.JUMPDEST + Op.JUMP(Op.PUSH0)
766+
jumps_address = pre.deploy_contract(jumps_code)
780767

781768
tx = Transaction(
782-
to=caller_address,
769+
to=jumps_address,
783770
gas_limit=env.gas_limit,
784771
sender=pre.fund_eoa(),
785772
)

0 commit comments

Comments
 (0)