PR feedback from @ericsson49: fix when `len(bytez) > BYTES_PER_CUSTOD…

…Y_ATOM`
ethereum · Jun 30, 2020 · 9663638 · 9663638
1 parent eaae70b
commit 9663638
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/specs/phase1/custody-game.md b/specs/phase1/custody-game.md
@@ -229,7 +229,8 @@ Given one set of data, return the custody atoms: each atom will be combined with
 
 ```python
 def get_custody_atoms(bytez: bytes) -> Sequence[bytes]:
-    bytez += b'\x00' * ((BYTES_PER_CUSTODY_ATOM - len(bytez)) % BYTES_PER_CUSTODY_ATOM)  # right-padding
+    length_remainder = len(bytez) % BYTES_PER_CUSTODY_ATOM
+    bytez += b'\x00' * ((BYTES_PER_CUSTODY_ATOM - length_remainder) % BYTES_PER_CUSTODY_ATOM)  # right-padding
     return [
         bytez[i:i + BYTES_PER_CUSTODY_ATOM]
         for i in range(0, len(bytez), BYTES_PER_CUSTODY_ATOM)