new(tests): EIP-2537 mainnet tests

Author: marioevz

tests/prague/eip2537_bls_12_381_precompiles/conftest.py

@@ -4,6 +4,7 @@
 
 import pytest
 
+from ethereum_test_forks import Fork
 from ethereum_test_tools import EOA, Address, Alloc, Bytecode, Storage, Transaction, keccak256
 from ethereum_test_tools import Opcodes as Op
 
@@ -155,7 +156,7 @@ def call_contract_address(pre: Alloc, call_contract_code: Bytecode) -> Address:
 @pytest.fixture
 def sender(pre: Alloc) -> EOA:
     """Sender of the transaction."""
-    return pre.fund_eoa(1_000_000_000_000_000_000)
+    return pre.fund_eoa()
 
 
 @pytest.fixture
@@ -169,9 +170,12 @@ def post(call_contract_address: Address, call_contract_post_storage: Storage):
 
 
 @pytest.fixture
-def tx_gas_limit(precompile_gas: int) -> int:
+def tx_gas_limit(fork: Fork, input_data: bytes, precompile_gas: int) -> int:
     """Transaction gas limit used for the test (Can be overridden in the test)."""
-    return 10_000_000 + precompile_gas
+    intrinsic_gas_cost_calc = fork.transaction_intrinsic_cost_calculator()
+    intrinsic_gas_cost = intrinsic_gas_cost_calc(calldata=input_data)
+    extra_gas = 100_000  # Memory expansion, sstores, return data, keccak, etc.
+    return intrinsic_gas_cost + extra_gas + precompile_gas
 
 
 @pytest.fixture

tests/prague/eip2537_bls_12_381_precompiles/test_eip.py

@@ -0,0 +1,91 @@
+"""
+abstract: Tests all precompiles of [EIP-2537: Precompile for BLS12-381 curve operations](https://eips.ethereum.org/EIPS/eip-2537)
+    Tests all precompiles of [EIP-2537: Precompile for BLS12-381 curve operations](https://eips.ethereum.org/EIPS/eip-2537).
+"""  # noqa: E501
+
+import pytest
+
+from ethereum_test_tools import Alloc, Environment, StateTestFiller, Transaction
+
+from .spec import FP, FP2, Scalar, Spec, ref_spec_2537
+
+REFERENCE_SPEC_GIT_PATH = ref_spec_2537.git_path
+REFERENCE_SPEC_VERSION = ref_spec_2537.version
+
+
+@pytest.mark.parametrize(
+    "precompile_address,input_data,expected_output,vector_gas_value",
+    [
+        pytest.param(
+            Spec.G1ADD,
+            Spec.G1 + Spec.INF_G1,
+            Spec.G1,
+            None,
+            id="G1ADD",
+        ),
+        pytest.param(
+            Spec.G1MSM,
+            Spec.G1 + Scalar(1) + Spec.INF_G1 + Scalar(1),
+            Spec.G1,
+            None,
+            id="G1MSM",
+        ),
+        pytest.param(
+            Spec.G2ADD,
+            Spec.G2 + Spec.INF_G2,
+            Spec.G2,
+            None,
+            id="G2ADD",
+        ),
+        pytest.param(
+            Spec.G2MSM,
+            Spec.G2 + Scalar(1) + Spec.INF_G2 + Scalar(1),
+            Spec.G2,
+            None,
+            id="G2MSM",
+        ),
+        pytest.param(
+            Spec.PAIRING,
+            Spec.G1 + Spec.INF_G2,
+            Spec.PAIRING_TRUE,
+            None,
+            id="PAIRING",
+        ),
+        pytest.param(
+            Spec.MAP_FP_TO_G1,
+            FP(
+                799950832265136997107648781861994410980648980263584507133499364313075404851459407870655748616451882783569609925573  # noqa: E501
+            ),
+            Spec.INF_G1,
+            None,
+            id="fp_map_to_inf",
+        ),
+        pytest.param(
+            Spec.MAP_FP2_TO_G2,
+            FP2(
+                (
+                    3510328712861478240121438855244276237335901234329585006107499559909114695366216070652508985150831181717984778988906,  # noqa: E501
+                    2924545590598115509050131525615277284817672420174395176262156166974132393611647670391999011900253695923948997972401,  # noqa: E501
+                )
+            ),
+            Spec.INF_G2,
+            None,
+            id="fp_map_to_inf",
+        ),
+    ],
+)
+@pytest.mark.valid_at("Prague")
+@pytest.mark.mainnet
+def test_eip_2537(
+    state_test: StateTestFiller,
+    pre: Alloc,
+    post: dict,
+    tx: Transaction,
+):
+    """Test the all precompiles of EIP-2537."""
+    state_test(
+        env=Environment(),
+        pre=pre,
+        tx=tx,
+        post=post,
+    )