fix

Author: felix314159

src/ethereum_test_benchmark/__init__.py

@@ -1,4 +1,7 @@
-"""Benchmark code generator classes for creating optimized bytecode patterns."""
+"""
+Benchmark code generator classes for
+creating optimized bytecode patterns.
+"""
 
 from .benchmark_code_generator import (
     BenchmarkCodeGenerator,

src/ethereum_test_benchmark/benchmark_code_generator.py

@@ -1,4 +1,7 @@
-"""Benchmark code generator classes for creating optimized bytecode patterns."""
+"""
+Benchmark code generator classes for creating
+optimized bytecode patterns.
+"""
 
 from ethereum_test_forks import Fork
 from ethereum_test_specs.benchmark import BenchmarkCodeGenerator
@@ -13,7 +16,8 @@ class JumpLoopGenerator(BenchmarkCodeGenerator):
     def deploy_contracts(self, pre: Alloc, fork: Fork) -> None:
         """Deploy the looping contract."""
         # Benchmark Test Structure:
-        # setup + JUMPDEST + attack + attack + ... + attack + JUMP(setup_length)
+        # setup + JUMPDEST + attack + attack + ... +
+        # attack + JUMP(setup_length)
         code = self.generate_repeated_code(self.attack_block, self.setup, fork)
         self._contract_address = pre.deploy_contract(code=code)
 
@@ -30,13 +34,17 @@ def generate_transaction(self, pre: Alloc, gas_limit: int, fork: Fork) -> Transa
 
 
 class ExtCallGenerator(BenchmarkCodeGenerator):
-    """Generates bytecode that fills the contract to maximum allowed code size."""
+    """
+    Generates bytecode that fills the contract to
+    maximum allowed code size.
+    """
 
     def deploy_contracts(self, pre: Alloc, fork: Fork) -> None:
         """Deploy both target and caller contracts."""
         # Benchmark Test Structure:
         # There are two contracts:
-        # 1. The target contract that executes certain operation but not loop (e.g. PUSH)
+        # 1. The target contract that executes certain operation
+        #    but not loop (e.g. PUSH)
         # 2. The loop contract that calls the target contract in a loop
 
         max_iterations = min(
@@ -49,8 +57,12 @@ def deploy_contracts(self, pre: Alloc, fork: Fork) -> None:
         )
 
         # Create caller contract that repeatedly calls the target contract
-        # attack = POP(STATICCALL(GAS, target_contract_address, 0, 0, 0, 0))
-        # setup + JUMPDEST + attack + attack + ... + attack + JUMP(setup_length)
+        # attack = POP(
+        #             STATICCALL(GAS, target_contract_address, 0, 0, 0, 0)
+        #          )
+        #
+        # setup + JUMPDEST + attack + attack + ... + attack +
+        # JUMP(setup_length)
         code_sequence = Op.POP(Op.STATICCALL(Op.GAS, self._target_contract_address, 0, 0, 0, 0))
 
         caller_code = self.generate_repeated_code(code_sequence, Bytecode(), fork)

src/ethereum_test_specs/benchmark.py

@@ -54,7 +54,10 @@ def generate_transaction(self, pre: Alloc, gas_limit: int, fork: Fork) -> Transa
     def generate_repeated_code(
         self, repeated_code: Bytecode, setup: Bytecode, fork: Fork
     ) -> Bytecode:
-        """Calculate the maximum number of iterations that can fit in the code size limit."""
+        """
+        Calculate the maximum number of iterations that
+        can fit in the code size limit.
+        """
         assert len(repeated_code) > 0, "repeated_code cannot be empty"
         max_code_size = fork.max_code_size()
 
@@ -114,7 +117,10 @@ class BenchmarkTest(BaseTest):
 
     @classmethod
     def pytest_parameter_name(cls) -> str:
-        """Return the parameter name used in pytest to select this spec type."""
+        """
+        Return the parameter name used in pytest
+        to select this spec type.
+        """
         return "benchmark_test"
 
     @classmethod
@@ -124,7 +130,10 @@ def discard_fixture_format_by_marks(
         fork: Fork,
         markers: List[pytest.Mark],
     ) -> bool:
-        """Discard a fixture format from filling if the appropriate marker is used."""
+        """
+        Discard a fixture format from filling if the
+        appropriate marker is used.
+        """
         if "blockchain_test_only" in [m.name for m in markers]:
             return fixture_format != BlockchainFixture
         if "blockchain_test_engine_only" in [m.name for m in markers]:
@@ -136,7 +145,10 @@ def get_genesis_environment(self, fork: Fork) -> Environment:
         return self.env
 
     def split_transaction(self, tx: Transaction, gas_limit_cap: int | None) -> List[Transaction]:
-        """Split a transaction that exceeds the gas limit cap into multiple transactions."""
+        """
+        Split a transaction that exceeds the gas
+        limit cap into multiple transactions.
+        """
         if gas_limit_cap is None:
             tx.gas_limit = HexNumber(self.gas_benchmark_value)
             return [tx]

src/ethereum_test_specs/tests/test_benchmark.py

@@ -1,4 +1,7 @@
-"""Tests for the BenchmarkTest class and its transaction splitting functionality."""
+"""
+Tests for the BenchmarkTest class and its
+transaction splitting functionality.
+"""
 
 import pytest
 
@@ -20,7 +23,10 @@
     ],
 )
 def test_split_transaction(gas_benchmark_value_millions: int, expected_splits: int):
-    """Test that transaction splitting works correctly for Osaka fork gas cap."""
+    """
+    Test that transaction splitting works
+    correctly for Osaka fork gas cap.
+    """
     gas_benchmark_value = gas_benchmark_value_millions * 1_000_000
     gas_limit_cap = 16_000_000  # Osaka's transaction gas limit cap
 
@@ -100,6 +106,7 @@ def test_split_transaction_edge_cases(gas_benchmark_value: int, gas_limit_cap: i
         # When no cap, gas_limit should be benchmark value
         assert split_txs[0].gas_limit == gas_benchmark_value
     else:
-        # When cap > benchmark, gas_limit should be min of tx.gas_limit and benchmark
+        # When cap > benchmark, gas_limit should be
+        # min of tx.gas_limit and benchmark
         assert benchmark_test.tx is not None, "Transaction should not be None"
         assert split_txs[0].gas_limit == min(benchmark_test.tx.gas_limit, gas_benchmark_value)

src/ethereum_test_vm/bytecode.py

@@ -245,7 +245,10 @@ def keccak256(self) -> Hash:
     def __get_pydantic_core_schema__(
         cls, source_type: Any, handler: GetCoreSchemaHandler
     ) -> PlainValidatorFunctionSchema:
-        """Provide Pydantic core schema for Bytecode serialization and validation."""
+        """
+        Provide Pydantic core schema for Bytecode
+        serialization and validation.
+        """
         return no_info_plain_validator_function(
             cls,
             serialization=plain_serializer_function_ser_schema(