Lint

Author: reedsa

tests/berlin/eip2929_gas_cost_increases/test_call_gas_cost_change.py

@@ -1,6 +1,6 @@
 """Tests supported precompiled contracts."""
 
-from typing import Iterator, Tuple
+from typing import Any, Iterator, Tuple
 
 import pytest
 
@@ -14,6 +14,7 @@
     Transaction,
 )
 from ethereum_test_tools.vm.opcode import Opcodes as Op
+from ethereum_test_types.types import EOA
 from ethereum_test_vm.bytecode import Bytecode
 
 REFERENCE_SPEC_GIT_PATH = "EIPS/eip-2929.md"
@@ -40,22 +41,22 @@ def contract_bytecode(**call_kwargs) -> Bytecode:
     )
 
 
-def initial_setup_bytecode(sender: Account) -> Bytecode:
+def initial_setup_bytecode(address: EOA) -> Bytecode:
     """Generate bytecode for the initial setup to be tested."""
     return (
-        Op.MSTORE(0, 0) + Op.MSTORE(0x100, 0xDEADBEEF) + Op.CALL(address=sender, value=1) + Op.POP
+        Op.MSTORE(0, 0) + Op.MSTORE(0x100, 0xDEADBEEF) + Op.CALL(address=address, value=1) + Op.POP
     )
 
 
-def call_addresses_and_expected_output(fork: Fork) -> Iterator[Tuple[str, bool]]:
+def call_addresses_and_expected_output(fork: Fork) -> Iterator[Any]:
     """
     Yield the addresses of precompiled contracts and their support status for a given fork.
 
     Args:
         fork (Fork): The fork instance containing precompiled contract information.
 
     Yields:
-        Iterator[Tuple[str, bool]]: A tuple containing the address in hexadecimal format and a
+        Iterator[Any]: A pytest ParameterSet containing the address in hexadecimal format and a
             boolean indicating whether the address is a supported precompile.
 
     """