logos-co
/
nomos-specs
mirror of https://github.com/logos-co/nomos-specs.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Update encode row test

This commit is contained in:
Daniel Sanchez Quiros 2024-03-05 18:17:28 +01:00
parent 17947ef74b
commit 0fa9f5edd2
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
da/test_encoder.py
View File

@ -14,7 +14,7 @@ class TestEncoder(TestCase):
def setUp(self): def setUp(self):
self.params: DAEncoderParams = DAEncoderParams(column_count=16, bytes_per_field_element=32) self.params: DAEncoderParams = DAEncoderParams(column_count=16, bytes_per_field_element=32)
self.encoder: DAEncoder = DAEncoder(self.params) self.encoder: DAEncoder = DAEncoder(self.params)
self.elements = 16 self.elements = 100
self.data = bytearray( self.data = bytearray(
chain.from_iterable( chain.from_iterable(
randrange(BLS_MODULUS).to_bytes(length=self.params.bytes_per_field_element, byteorder='big') randrange(BLS_MODULUS).to_bytes(length=self.params.bytes_per_field_element, byteorder='big')
@ -49,6 +49,7 @@ class TestEncoder(TestCase):
chunks_matrix = self.encoder._chunkify_data(self.data) chunks_matrix = self.encoder._chunkify_data(self.data)
extended_chunks_matrix = self.encoder._rs_encode_rows(chunks_matrix) extended_chunks_matrix = self.encoder._rs_encode_rows(chunks_matrix)
for r1, r2 in zip(chunks_matrix, extended_chunks_matrix): for r1, r2 in zip(chunks_matrix, extended_chunks_matrix):
self.assertEqual(len(r1), len(r2)//2)
r2 = [BLSFieldElement.from_bytes(x) for x in batched(r2, self.params.bytes_per_field_element)] r2 = [BLSFieldElement.from_bytes(x) for x in batched(r2, self.params.bytes_per_field_element)]
poly_1 = kzg.bytes_to_polynomial(r1) poly_1 = kzg.bytes_to_polynomial(r1)
# we check against decoding so we now the encoding was properly done # we check against decoding so we now the encoding was properly done