98 lines
2.2 KiB
Nim
98 lines
2.2 KiB
Nim
import math, random, strformat, times, std/monotimes
|
|
import fixedseq, game, simulation, ui
|
|
|
|
|
|
type
|
|
TestResults = object
|
|
ops: int
|
|
time: Duration
|
|
|
|
|
|
proc summarize(tr: TestResults) =
|
|
let secs = tr.time.inMilliseconds.float / 1000
|
|
stdout.write("Test completed:\n")
|
|
stdout.write(" " & $tr.ops, " operations in " & $round(secs, 2) & " seconds\n")
|
|
stdout.write(" " & $round(tr.ops.float / secs, 2) & " operations per second")
|
|
stdout.flushFile()
|
|
|
|
|
|
template executionTime(body: untyped): Duration =
|
|
let start = getMonoTime()
|
|
body
|
|
getMonoTime() - start
|
|
|
|
|
|
proc getRand(): Rand =
|
|
randomize()
|
|
result = initRand(rand(int64))
|
|
|
|
|
|
proc randomDice(r: var Rand): seq[tuple[c: Color, p: int]] =
|
|
for c in Color:
|
|
let v = r.rand(1..3)
|
|
result.add((c, v))
|
|
result.shuffle
|
|
|
|
|
|
proc newRandomGame(r: var Rand): Board =
|
|
var dice: array[5, tuple[c: Color, p: int]]
|
|
for i in 0 .. 4:
|
|
dice[i] = (Color(i), r.rand(1..3))
|
|
|
|
result.init
|
|
result.setState(dice, [])
|
|
|
|
|
|
proc games(nTests, nSamples: SomeInteger, parallel = true): TestResults =
|
|
var r = getRand()
|
|
var scores: ScoreSet
|
|
for i in 1 .. nTests:
|
|
let b = newRandomGame(r)
|
|
let dur = executionTime:
|
|
let s = b.randomGames(nSamples, parallel = parallel)
|
|
result.ops += s.sum()
|
|
result.time += dur
|
|
|
|
|
|
proc testLegs(n: Natural = 100): auto =
|
|
var boards: seq[Board]
|
|
var r = initRand(rand(int64))
|
|
for i in 1 .. n:
|
|
var b: Board
|
|
b.init
|
|
let dice = randomDice(r)
|
|
b.setState(dice, [])
|
|
boards.add(b)
|
|
stdout.write("Constructed: " & $i & "\r")
|
|
echo ""
|
|
|
|
echo "Running..."
|
|
let start = cpuTime()
|
|
for b in boards:
|
|
discard b.getLegScores
|
|
result = cpuTime() - start
|
|
|
|
|
|
proc testSpread(nTests, nSamples: Natural) =
|
|
var b: Board
|
|
b.init
|
|
var r = initRand(rand(int64))
|
|
let dice = randomDice(r)
|
|
b.setState(dice, [])
|
|
b.display(1, 5)
|
|
let spread = randomSpread(b, nTests, nSamples)
|
|
|
|
stdout.writeLine("Variance:")
|
|
for c in Color:
|
|
let variance = 100 * (spread.hi[c] - spread.lo[c])
|
|
stdout.writeLine(fmt"{c}: {round(variance, 2):.2f}%")
|
|
|
|
let diff = 100 * (max(spread.hi) - min(spread.lo))
|
|
stdout.writeLine(fmt"Win percentage differential: {round(diff, 2):.2f}%")
|
|
|
|
stdout.flushFile()
|
|
|
|
|
|
when isMainModule:
|
|
games(10, 10_000_000).summarize()
|