c-kzg-4844/bindings/python/tests.py

35 lines
816 B
Python

import ckzg
import random
# Commit to a few random blobs
BLOB_SIZE = 4096
MAX_BLOBS_PER_BLOCK = 16
blobs = [
random.randbytes(32 * BLOB_SIZE) for _ in range(3)
]
ts = ckzg.load_trusted_setup("../../src/trusted_setup.txt")
kzg_commitments = [ckzg.blob_to_kzg_commitment(blob, ts) for blob in blobs]
# Compute proof for these blobs
blobs_bytes = b''.join(blobs)
proof = ckzg.compute_aggregate_kzg_proof(blobs_bytes, ts)
# Verify proof
assert ckzg.verify_aggregate_kzg_proof(blobs_bytes, kzg_commitments, proof, ts), 'verify failed'
# Verification fails at wrong value
other = b'x' if not blobs_bytes.endswith(b'x') else b'y'
other_bytes = blobs_bytes[:-1] + other
assert not ckzg.verify_aggregate_kzg_proof(other_bytes, kzg_commitments, proof, ts), 'verify succeeded incorrectly'
print('tests passed')