tzhash/tz/sl2_test.go

65 lines
1,008 B
Go
Raw Normal View History

2018-12-29 13:04:17 +00:00
package tz
import (
"math/rand"
"testing"
"time"
"github.com/nspcc-dev/tzhash/gf127"
. "github.com/onsi/gomega"
)
func init() {
rand.Seed(time.Now().UnixNano())
}
func random() (a *sl2) {
a = new(sl2)
a[0][0] = *gf127.Random()
a[0][1] = *gf127.Random()
a[1][0] = *gf127.Random()
// so that result is in SL2
// d = a^-1*(1+b*c)
gf127.Mul(&a[0][1], &a[1][0], &a[1][1])
gf127.Add(&a[1][1], gf127.New(1, 0), &a[1][1])
t := gf127.New(0, 0)
gf127.Inv(&a[0][0], t)
gf127.Mul(t, &a[1][1], &a[1][1])
return
2018-12-29 13:04:17 +00:00
}
func TestSL2_MarshalBinary(t *testing.T) {
var (
a = random()
b = new(sl2)
g = NewGomegaWithT(t)
)
2018-12-29 13:04:17 +00:00
data, err := a.MarshalBinary()
g.Expect(err).NotTo(HaveOccurred())
err = b.UnmarshalBinary(data)
g.Expect(err).NotTo(HaveOccurred())
g.Expect(a).To(Equal(b))
}
func TestInv(t *testing.T) {
var (
a, b, c *sl2
g = NewGomegaWithT(t)
)
c = new(sl2)
for i := 0; i < 5; i++ {
a = random()
b = Inv(a)
c = c.Mul(a, b)
g.Expect(*c).To(Equal(id))
}
}