Port kzg_4844 tests to pytest

tests/core/pyspec/eth2spec/test/deneb/kzg/test_compute_blob_kzg_proof.py

@@ -0,0 +1,92 @@
+###############################################################################
+# Test cases for compute_blob_kzg_proof
+###############################################################################
+
+from eth_utils.hexadecimal import encode_hex
+
+from eth2spec.test.context import only_generator, single_phase, spec_test, with_phases
+from eth2spec.test.helpers.constants import DENEB
+from eth2spec.test.utils.kzg_tests import G1, INVALID_BLOBS, INVALID_G1_POINTS, VALID_BLOBS
+from tests.infra.manifest import manifest
+from tests.infra.template_test import template_test
+
+
+def _run_compute_blob_kzg_proof_test(spec, blob, commitment, valid: bool):
+    try:
+        proof = spec.compute_blob_kzg_proof(blob, commitment)
+    except Exception:
+        proof = None
+
+    assert (proof is not None) == valid
+
+    yield (
+        "data",
+        "data",
+        {
+            "input": {
+                "blob": encode_hex(blob),
+                "commitment": encode_hex(commitment),
+            },
+            "output": encode_hex(proof) if proof is not None else None,
+        },
+    )
+
+
+@template_test
+def _compute_blob_kzg_proof_case_valid_blob(blob_index):
+    blob = VALID_BLOBS[blob_index]
+
+    @manifest(preset_name="general", suite_name="kzg-mainnet")
+    @only_generator("too slow")
+    @with_phases([DENEB])
+    @spec_test
+    @single_phase
+    def the_test(spec):
+        commitment = spec.blob_to_kzg_commitment(blob)
+        yield from _run_compute_blob_kzg_proof_test(spec, blob, commitment, valid=True)
+
+    return (the_test, f"test_compute_blob_kzg_proof_case_valid_blob_{blob_index}")
+
+
+for blob_index in range(len(VALID_BLOBS)):
+    _compute_blob_kzg_proof_case_valid_blob(blob_index)
+
+
+@template_test
+def _compute_blob_kzg_proof_case_invalid_blob(blob_index):
+    blob = INVALID_BLOBS[blob_index]
+    commitment = G1
+
+    @manifest(preset_name="general", suite_name="kzg-mainnet")
+    @only_generator("too slow")
+    @with_phases([DENEB])
+    @spec_test
+    @single_phase
+    def the_test(spec):
+        yield from _run_compute_blob_kzg_proof_test(spec, blob, commitment, valid=False)
+
+    return (the_test, f"test_compute_blob_kzg_proof_case_invalid_blob_{blob_index}")
+
+
+for blob_index in range(len(INVALID_BLOBS)):
+    _compute_blob_kzg_proof_case_invalid_blob(blob_index)
+
+
+@template_test
+def _compute_blob_kzg_proof_case_invalid_commitment(commitment_index):
+    blob = VALID_BLOBS[1]
+    commitment = INVALID_G1_POINTS[commitment_index]
+
+    @manifest(preset_name="general", suite_name="kzg-mainnet")
+    @only_generator("too slow")
+    @with_phases([DENEB])
+    @spec_test
+    @single_phase
+    def the_test(spec):
+        yield from _run_compute_blob_kzg_proof_test(spec, blob, commitment, valid=False)
+
+    return (the_test, f"test_compute_blob_kzg_proof_case_invalid_commitment_{commitment_index}")
+
+
+for commitment_index in range(len(INVALID_G1_POINTS)):
+    _compute_blob_kzg_proof_case_invalid_commitment(commitment_index)

tests/core/pyspec/eth2spec/test/deneb/kzg/test_compute_challenge.py

@@ -0,0 +1,70 @@
+###############################################################################
+# Test cases for compute_challenge
+###############################################################################
+
+from eth_utils.hexadecimal import encode_hex
+
+from eth2spec.test.context import only_generator, single_phase, spec_test, with_phases
+from eth2spec.test.helpers.constants import DENEB
+from eth2spec.test.utils.kzg_tests import VALID_BLOBS
+from tests.infra.manifest import manifest
+from tests.infra.template_test import template_test
+
+
+def _run_compute_challenge_test(spec, blob, commitment):
+    challenge = spec.compute_challenge(blob, commitment)
+
+    yield (
+        "data",
+        "data",
+        {
+            "input": {
+                "blob": encode_hex(blob),
+                "commitment": encode_hex(commitment),
+            },
+            "output": encode_hex(spec.bls_field_to_bytes(challenge)),
+        },
+    )
+
+
+@template_test
+def _compute_challenge_case_valid(blob_index):
+    blob = VALID_BLOBS[blob_index]
+
+    @manifest(preset_name="general", suite_name="kzg-mainnet")
+    @only_generator("too slow")
+    @with_phases([DENEB])
+    @spec_test
+    @single_phase
+    def the_test(spec):
+        commitment = spec.blob_to_kzg_commitment(blob)
+        yield from _run_compute_challenge_test(spec, blob, commitment)
+
+    return (the_test, f"test_compute_challenge_case_valid_{blob_index}")
+
+
+for blob_index in range(len(VALID_BLOBS)):
+    _compute_challenge_case_valid(blob_index)
+
+
+@manifest(preset_name="general", suite_name="kzg-mainnet")
+@only_generator("too slow")
+@with_phases([DENEB])
+@spec_test
+@single_phase
+def test_compute_challenge_case_mismatched_commitment(spec):
+    # Use commitment from a different blob
+    commitment = spec.blob_to_kzg_commitment(VALID_BLOBS[4])
+    blob = VALID_BLOBS[3]
+    yield from _run_compute_challenge_test(spec, blob, commitment)
+
+
+@manifest(preset_name="general", suite_name="kzg-mainnet")
+@only_generator("too slow")
+@with_phases([DENEB])
+@spec_test
+@single_phase
+def test_compute_challenge_case_commitment_at_infinity(spec):
+    commitment = spec.G1_POINT_AT_INFINITY
+    blob = VALID_BLOBS[4]
+    yield from _run_compute_challenge_test(spec, blob, commitment)

tests/core/pyspec/eth2spec/test/deneb/kzg/test_compute_kzg_proof.py

@@ -0,0 +1,99 @@
+###############################################################################
+# Test cases for compute_kzg_proof
+###############################################################################
+
+from eth_utils import encode_hex
+
+from eth2spec.test.context import only_generator, single_phase, spec_test, with_phases
+from eth2spec.test.helpers.constants import DENEB
+from eth2spec.test.utils.kzg_tests import (
+    INVALID_BLOBS,
+    INVALID_FIELD_ELEMENTS,
+    VALID_BLOBS,
+    VALID_FIELD_ELEMENTS,
+)
+from tests.infra.manifest import manifest
+from tests.infra.template_test import template_test
+
+
+def _run_compute_kzg_proof_test(spec, blob, z, valid: bool):
+    try:
+        proof, y = spec.compute_kzg_proof(blob, z)
+        result = (encode_hex(proof), encode_hex(y))
+    except Exception:
+        result = None
+
+    assert (result is not None) == valid
+
+    yield (
+        "data",
+        "data",
+        {
+            "input": {
+                "blob": encode_hex(blob),
+                "z": encode_hex(z),
+            },
+            "output": result,
+        },
+    )
+
+
+@template_test
+def _compute_kzg_proof_case_valid(blob_index, z_index):
+    blob = VALID_BLOBS[blob_index]
+    z = VALID_FIELD_ELEMENTS[z_index]
+
+    @manifest(preset_name="general", suite_name="kzg-mainnet")
+    @only_generator("too slow")
+    @with_phases([DENEB])
+    @spec_test
+    @single_phase
+    def the_test(spec):
+        yield from _run_compute_kzg_proof_test(spec, blob, z, valid=True)
+
+    return (the_test, f"test_compute_kzg_proof_case_valid_blob_{blob_index}_{z_index}")
+
+
+for blob_index in range(len(VALID_BLOBS)):
+    for z_index in range(len(VALID_FIELD_ELEMENTS)):
+        _compute_kzg_proof_case_valid(blob_index, z_index)
+
+
+@template_test
+def _compute_kzg_proof_case_invalid_blob(blob_index):
+    blob = INVALID_BLOBS[blob_index]
+    z = VALID_FIELD_ELEMENTS[0]
+
+    @manifest(preset_name="general", suite_name="kzg-mainnet")
+    @only_generator("too slow")
+    @with_phases([DENEB])
+    @spec_test
+    @single_phase
+    def the_test(spec):
+        yield from _run_compute_kzg_proof_test(spec, blob, z, valid=False)
+
+    return (the_test, f"test_compute_kzg_proof_case_invalid_blob_{blob_index}")
+
+
+for blob_index in range(len(INVALID_BLOBS)):
+    _compute_kzg_proof_case_invalid_blob(blob_index)
+
+
+@template_test
+def _compute_kzg_proof_case_invalid_z(z_index):
+    blob = VALID_BLOBS[4]
+    z = INVALID_FIELD_ELEMENTS[z_index]
+
+    @manifest(preset_name="general", suite_name="kzg-mainnet")
+    @only_generator("too slow")
+    @with_phases([DENEB])
+    @spec_test
+    @single_phase
+    def the_test(spec):
+        yield from _run_compute_kzg_proof_test(spec, blob, z, valid=False)
+
+    return (the_test, f"test_compute_kzg_proof_case_invalid_z_{z_index}")
+
+
+for z_index in range(len(INVALID_FIELD_ELEMENTS)):
+    _compute_kzg_proof_case_invalid_z(z_index)