40 lines
1.4 KiB
Nim
40 lines
1.4 KiB
Nim
# beacon_chain
|
|
# Copyright (c) 2018 Status Research & Development GmbH
|
|
# Licensed and distributed under either of
|
|
# * MIT license (license terms in the root directory or at http://opensource.org/licenses/MIT).
|
|
# * Apache v2 license (license terms in the root directory or at http://www.apache.org/licenses/LICENSE-2.0).
|
|
# at your option. This file may not be copied, modified, or distributed except according to those terms.
|
|
|
|
import
|
|
# Standard library
|
|
ospaths, strutils, unittest,
|
|
# Beacon chain internals
|
|
../../beacon_chain/spec/[datatypes, validator, digest],
|
|
# Test utilities
|
|
../testutil,
|
|
./fixtures_utils
|
|
|
|
type
|
|
Shuffling* = object
|
|
seed*: Eth2Digest
|
|
count*: uint64
|
|
shuffled*: seq[ValidatorIndex]
|
|
|
|
const TestFolder = currentSourcePath.rsplit(DirSep, 1)[0]
|
|
|
|
when const_preset == "mainnet":
|
|
const TestsPath = "fixtures" / "json_tests" / "shuffling" / "core" / "shuffling_full.json"
|
|
elif const_preset == "minimal":
|
|
const TestsPath = "fixtures" / "json_tests" / "shuffling" / "core" / "shuffling_minimal.json"
|
|
|
|
var shufflingTests: Tests[Shuffling]
|
|
|
|
suite "Official - Shuffling tests [Preset: " & preset():
|
|
test "Parsing the official shuffling tests [Preset: " & preset():
|
|
shufflingTests = parseTests(TestFolder / TestsPath, Shuffling)
|
|
|
|
test "Shuffling a sequence of N validators" & preset():
|
|
for t in shufflingTests.test_cases:
|
|
let implResult = get_shuffled_seq(t.seed, t.count)
|
|
check: implResult == t.shuffled
|