ceremonyclient/vdf/vdf_test.go

54 lines
1.3 KiB
Go
Raw Normal View History

2024-06-08 11:32:45 +00:00
package vdf_test
import (
"golang.org/x/crypto/sha3"
nekrovdf "source.quilibrium.com/quilibrium/monorepo/nekryptology/pkg/vdf"
"source.quilibrium.com/quilibrium/monorepo/vdf"
"testing"
)
func getChallenge(seed string) [32]byte {
return sha3.Sum256([]byte(seed))
}
func TestProveVerify(t *testing.T) {
difficulty := uint32(10000)
challenge := getChallenge("TestProveVerify")
solution := vdf.WesolowskiSolve(challenge, difficulty)
isOk := vdf.WesolowskiVerify(challenge, difficulty, solution)
if !isOk {
t.Fatalf("Verification failed")
}
}
func TestProveRustVerifyNekro(t *testing.T) {
difficulty := uint32(100)
challenge := getChallenge("TestProveRustVerifyNekro")
for i := 0; i < 100; i++ {
solution := vdf.WesolowskiSolve(challenge, difficulty)
nekroVdf := nekrovdf.New(difficulty, challenge)
isOk := nekroVdf.Verify(solution)
if !isOk {
t.Fatalf("Verification failed")
}
challenge = sha3.Sum256(solution[:])
}
}
func TestProveNekroVerifyRust(t *testing.T) {
difficulty := uint32(100)
challenge := getChallenge("TestProveNekroVerifyRust")
for i := 0; i < 100; i++ {
nekroVdf := nekrovdf.New(difficulty, challenge)
nekroVdf.Execute()
proof := nekroVdf.GetOutput()
isOk := vdf.WesolowskiVerify(challenge, difficulty, proof)
if !isOk {
t.Fatalf("Verification failed")
}
challenge = sha3.Sum256(proof[:])
}
}