mirror of
https://github.com/codex-storage/nim-poseidon2.git
synced 2025-02-24 16:58:11 +00:00
dbe3d8a692
For two reasons: - to distinguish them from the marshalling functions from constantine - they do not follow the convention that something that is marshalled can be unmarshalled, because they take in 31 bytes but produce 32 bytes
42 lines
1.6 KiB
Nim
42 lines
1.6 KiB
Nim
import std/unittest
|
|
import std/sequtils
|
|
import constantine/math/io/io_bigints
|
|
import constantine/math/arithmetic
|
|
import poseidon2/types
|
|
import poseidon2/io
|
|
|
|
suite "conversion to/from bytes":
|
|
|
|
test "converts little endian bytes into field elements":
|
|
let bytes = toSeq 1'u8..31'u8
|
|
let paddedTo32 = bytes & @[0'u8] # most significant byte is not used
|
|
let expected = F.fromBig(B.unmarshal(paddedTo32, littleEndian))
|
|
let unmarshalled = F.fromBytes(bytes)
|
|
check bool(unmarshalled == expected)
|
|
|
|
test "pads little endian bytes to the right with 0's":
|
|
let bytes = @[0x56'u8, 0x34, 0x12]
|
|
let paddedTo32 = bytes & 0'u8.repeat(32 - bytes.len)
|
|
let expected = F.fromBig(B.unmarshal(paddedTo32, littleEndian))
|
|
let unmarshalled = F.fromBytes(bytes)
|
|
check bool(unmarshalled == expected)
|
|
|
|
test "converts every 31 bytes into a field element":
|
|
let bytes = toSeq 1'u8..80'u8
|
|
let padded = bytes & 0'u8.repeat(93 - bytes.len)
|
|
let expected1 = F.fromBig(B.unmarshal(padded[ 0..<31] & @[0'u8], littleEndian))
|
|
let expected2 = F.fromBig(B.unmarshal(padded[31..<62] & @[0'u8], littleEndian))
|
|
let expected3 = F.fromBig(B.unmarshal(padded[62..<93] & @[0'u8], littleEndian))
|
|
let elements = seq[F].fromBytes(bytes)
|
|
check elements.len == 3
|
|
check bool(elements[0] == expected1)
|
|
check bool(elements[1] == expected2)
|
|
check bool(elements[2] == expected3)
|
|
|
|
test "converts field element into little-endian bytes":
|
|
var element: F
|
|
setMinusOne(element) # largest element in the field
|
|
var expected: array[32, byte]
|
|
marshal(expected, element.toBig(), littleEndian)
|
|
check element.toBytes() == expected
|