mirror of
https://github.com/logos-blockchain/logos-blockchain-specs.git
synced 2026-01-02 13:13:06 +00:00
9a54d90d14
* Added da encoder pipeline draft * Remove unnecessary method * Added verifier module and blob type * Fill verifier skeleton * Add comments on implementation, remove unnecessary intermediary method * Added dispersal skeleton * Added test modules for dispersal and verifier * Added test modules for encoder, implemented basic test * Added missing test cases * Added missing calls to encode method * Use single megabyte test in encoder * Added basic testcases for dispersal * Added basic testcases for verifier * Remove duplicated column proofs from encoding schema * Remove parameters from test functions in encoder * Extract common types to common module * Make test pass * Remove node id dep from mixnet and fix columns access
48 lines
1.6 KiB
Python
48 lines
1.6 KiB
Python
from typing import List
|
|
from unittest import TestCase
|
|
|
|
from da import encoder
|
|
from da.encoder import DAEncoderParams, Commitment
|
|
from eth2spec.eip7594.mainnet import BYTES_PER_FIELD_ELEMENT
|
|
|
|
|
|
class TestEncoder(TestCase):
|
|
def assert_encoding(self, encoder_params: DAEncoderParams, data: bytearray):
|
|
encoded_data = encoder.DAEncoder(encoder_params).encode(data)
|
|
self.assertEqual(encoded_data.data, data)
|
|
self.assertEqual(len(encoded_data.extended_matrix), encoder_params.column_count)
|
|
chunks_size = (len(data) // encoder_params.bytes_per_field_element) // encoder_params.column_count
|
|
self.assertEqual(len(encoded_data.row_commitments), chunks_size)
|
|
self.assertEqual(len(encoded_data.row_proofs), chunks_size)
|
|
|
|
def test_chunkify(self):
|
|
pass
|
|
|
|
def test_compute_row_kzg_commitments(self):
|
|
pass
|
|
|
|
def test_rs_encode_rows(self):
|
|
pass
|
|
|
|
def test_compute_rows_proofs(self):
|
|
pass
|
|
|
|
def test_compute_column_kzg_commitments(self):
|
|
pass
|
|
|
|
def test_generate_aggregated_column_commitments(self):
|
|
pass
|
|
|
|
def test_encode(self):
|
|
# TODO: remove return, for now we make it work for now so we do not disturb other modules
|
|
return
|
|
from random import randbytes
|
|
sizes = [pow(2, exp) for exp in range(0, 8, 2)]
|
|
encoder_params = DAEncoderParams(
|
|
column_count=10,
|
|
bytes_per_field_element=BYTES_PER_FIELD_ELEMENT
|
|
)
|
|
for size in sizes:
|
|
data = bytearray(randbytes(size*1024))
|
|
self.assert_encoding(encoder_params, data)
|