mirror of
https://github.com/pseXperiments/icicle.git
synced 2026-01-09 13:07:59 -05:00
49 lines
1.2 KiB
Go
49 lines
1.2 KiB
Go
package tests
|
|
|
|
import (
|
|
"os"
|
|
"testing"
|
|
|
|
"github.com/ingonyama-zk/icicle/v2/wrappers/golang/core"
|
|
bls12_381 "github.com/ingonyama-zk/icicle/v2/wrappers/golang/curves/bls12381"
|
|
ntt "github.com/ingonyama-zk/icicle/v2/wrappers/golang/curves/bls12381/ntt"
|
|
poly "github.com/ingonyama-zk/icicle/v2/wrappers/golang/curves/bls12381/polynomial"
|
|
|
|
"github.com/consensys/gnark-crypto/ecc/bls12-381/fr/fft"
|
|
)
|
|
|
|
const (
|
|
largestTestSize = 20
|
|
)
|
|
|
|
func initDomain[T any](largestTestSize int, cfg core.NTTConfig[T]) core.IcicleError {
|
|
rouMont, _ := fft.Generator(uint64(1 << largestTestSize))
|
|
rou := rouMont.Bits()
|
|
rouIcicle := bls12_381.ScalarField{}
|
|
limbs := core.ConvertUint64ArrToUint32Arr(rou[:])
|
|
|
|
rouIcicle.FromLimbs(limbs)
|
|
e := ntt.InitDomain(rouIcicle, cfg.Ctx, false)
|
|
return e
|
|
}
|
|
|
|
func TestMain(m *testing.M) {
|
|
poly.InitPolyBackend()
|
|
|
|
// setup domain
|
|
cfg := ntt.GetDefaultNttConfig()
|
|
e := initDomain(largestTestSize, cfg)
|
|
if e.IcicleErrorCode != core.IcicleErrorCode(0) {
|
|
panic("initDomain failed")
|
|
}
|
|
|
|
// execute tests
|
|
os.Exit(m.Run())
|
|
|
|
// release domain
|
|
e = ntt.ReleaseDomain(cfg.Ctx)
|
|
if e.IcicleErrorCode != core.IcicleErrorCode(0) {
|
|
panic("ReleaseDomain failed")
|
|
}
|
|
}
|