status-im
/
nimbus-eth2
mirror of https://github.com/status-im/nimbus-eth2.git
2
0
Fork 0
Code Issues Projects Releases Wiki Activity

add BitVector [skip ci]

This commit is contained in:
Mamy André-Ratsimbazafy 2019-11-05 17:53:00 +01:00 committed by zah
parent 3b8fc8fab8
commit 1caea64ac9
1 changed files with 25 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

26
tests/official/test_fixture_ssz_generic_types.nim
View File

@ -10,7 +10,7 @@ import
os, unittest, strutils, streams, strformat, strscans,
macros,
# Status libraries
stint,
stint, stew/bitseqs,
# Third-party
yaml,
# Beacon chain internals
@ -106,6 +106,29 @@ proc checkVector(sszSubType, dir: string, expectedHash: SSZHashTreeRoot) =
echo &" (SSZ) Vector[{typeIdent:7}, {size:3}]"
testVector(typeIdent, size)
proc testBitVector(size:static int, dir: string, expectedHash: SSZHashTreeRoot) =
let deserialized = SSZ.loadFile(dir/"serialized.ssz", BitArray[size])
check:
expectedHash.root == "0x" & toLowerASCII($deserialized.hashTreeRoot())
# TODO check the value
proc checkBitVector(sszSubType, dir: string, expectedHash: SSZHashTreeRoot) =
var size: int
let wasMatched = scanf(sszSubType, "bitvec_$i", size)
case size
of 1: testBitVector(1, dir, expectedHash)
of 2: testBitVector(2, dir, expectedHash)
of 3: testBitVector(3, dir, expectedHash)
of 4: testBitVector(4, dir, expectedHash)
of 5: testBitVector(5, dir, expectedHash)
of 8: testBitVector(8, dir, expectedHash)
of 16: testBitVector(16, dir, expectedHash)
of 31: testBitVector(31, dir, expectedHash)
of 512: testBitVector(512, dir, expectedHash)
of 513: testBitVector(513, dir, expectedHash)
else:
raise newException(ValueError, "Unsupported BitVector test size: " & $size)
# Test dispatch for valid inputs
# ------------------------------------------------------------------------
@ -135,6 +158,7 @@ proc sszCheck(sszType, sszSubType: string) =
else:
raise newException(ValueError, "unknown uint in test: " & sszSubType)
of "basic_vector": checkVector(sszSubType, dir, expectedHash)
of "bit_vector": checkBitVector(sszSubType, dir, expectedHash)
else:
discard # TODO