mirror of
https://github.com/luxfi/threshold.git
synced 2026-10-30 15:15:53 +00:00
zeekay
eeaa2a1c07
- Fix go vet error in lss_bench_test.go: replace b.Fatal with panic in goroutine - Fix go vet error in lss_test.go: remove t.Skip call from goroutine - Apply gofmt to all files to fix formatting issues All go vet errors and formatting issues have been resolved.
87 lines
2.3 KiB
Go
87 lines
2.3 KiB
Go
package polynomial
|
|
|
|
import (
|
|
"crypto/rand"
|
|
"fmt"
|
|
"testing"
|
|
|
|
"github.com/fxamacker/cbor/v2"
|
|
"github.com/luxfi/threshold/pkg/math/curve"
|
|
"github.com/luxfi/threshold/pkg/math/sample"
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestExponent_Evaluate(t *testing.T) {
|
|
group := curve.Secp256k1{}
|
|
|
|
var lhs curve.Point
|
|
for x := 0; x < 5; x++ {
|
|
N := 1000
|
|
secret := group.NewScalar()
|
|
if x%2 == 0 {
|
|
secret = sample.Scalar(rand.Reader, group)
|
|
}
|
|
poly := NewPolynomial(group, N, secret)
|
|
polyExp := NewPolynomialExponent(poly)
|
|
|
|
randomIndex := sample.Scalar(rand.Reader, group)
|
|
|
|
lhs = poly.Evaluate(randomIndex).ActOnBase()
|
|
rhs1 := polyExp.Evaluate(randomIndex)
|
|
rhs2 := polyExp.evaluateClassic(randomIndex)
|
|
|
|
require.Truef(t, lhs.Equal(rhs1), fmt.Sprint("base eval differs from horner", x))
|
|
require.Truef(t, lhs.Equal(rhs2), fmt.Sprint("base eval differs from classic", x))
|
|
require.Truef(t, rhs1.Equal(rhs2), fmt.Sprint("horner differs from classic", x))
|
|
}
|
|
}
|
|
|
|
func TestSum(t *testing.T) {
|
|
group := curve.Secp256k1{}
|
|
|
|
N := 20
|
|
Deg := 10
|
|
|
|
randomIndex := sample.Scalar(rand.Reader, group)
|
|
|
|
// compute f1(x) + f2(x) + …
|
|
evaluationScalar := group.NewScalar()
|
|
|
|
// compute F1(x) + F2(x) + …
|
|
evaluationPartial := group.NewPoint()
|
|
|
|
polys := make([]*Polynomial, N)
|
|
polysExp := make([]*Exponent, N)
|
|
for i := range polys {
|
|
sec := sample.Scalar(rand.Reader, group)
|
|
polys[i] = NewPolynomial(group, Deg, sec)
|
|
polysExp[i] = NewPolynomialExponent(polys[i])
|
|
|
|
evaluationScalar.Add(polys[i].Evaluate(randomIndex))
|
|
evaluationPartial = evaluationPartial.Add(polysExp[i].Evaluate(randomIndex))
|
|
}
|
|
|
|
// compute (F1 + F2 + …)(x)
|
|
summedExp, _ := Sum(polysExp)
|
|
evaluationSum := summedExp.Evaluate(randomIndex)
|
|
|
|
evaluationFromScalar := evaluationScalar.ActOnBase()
|
|
assert.True(t, evaluationSum.Equal(evaluationFromScalar))
|
|
assert.True(t, evaluationSum.Equal(evaluationPartial))
|
|
}
|
|
|
|
func TestMarshall(t *testing.T) {
|
|
group := curve.Secp256k1{}
|
|
|
|
sec := sample.Scalar(rand.Reader, group)
|
|
poly := NewPolynomial(group, 10, sec)
|
|
polyExp := NewPolynomialExponent(poly)
|
|
out, err := cbor.Marshal(polyExp)
|
|
require.NoError(t, err, "failed to Marshal")
|
|
polyExp2 := EmptyExponent(group)
|
|
err = cbor.Unmarshal(out, polyExp2)
|
|
require.NoError(t, err, "failed to Unmarshal")
|
|
assert.True(t, polyExp.Equal(*polyExp2), "should be the same")
|
|
}
|