forked from TrueCloudLab/tzhash
39 lines
660 B
Go
39 lines
660 B
Go
|
package tz
|
||
|
|
|
||
|
|
import (
|
||
|
|
"math"
|
||
|
|
"math/rand"
|
||
|
|
"testing"
|
||
|
|
"time"
|
||
|
|
|
||
|
|
"github.com/nspcc-dev/tzhash/gf127"
|
||
|
|
. "github.com/onsi/gomega"
|
||
|
|
)
|
||
|
|
|
||
|
|
func init() {
|
||
|
|
rand.Seed(time.Now().UnixNano())
|
||
|
|
}
|
||
|
|
|
||
|
|
func u64() uint64 {
|
||
|
|
return rand.Uint64() & (math.MaxUint64 >> 1)
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestSL2_MarshalBinary(t *testing.T) {
|
||
|
|
g := NewGomegaWithT(t)
|
||
|
|
|
||
|
|
a := new(sl2)
|
||
|
|
a[0][0] = *gf127.New(u64(), u64())
|
||
|
|
a[0][1] = *gf127.New(u64(), u64())
|
||
|
|
a[1][0] = *gf127.New(u64(), u64())
|
||
|
|
a[1][1] = *gf127.New(u64(), u64())
|
||
|
|
|
||
|
|
data, err := a.MarshalBinary()
|
||
|
|
g.Expect(err).NotTo(HaveOccurred())
|
||
|
|
|
||
|
|
b := new(sl2)
|
||
|
|
err = b.UnmarshalBinary(data)
|
||
|
|
g.Expect(err).NotTo(HaveOccurred())
|
||
|
|
|
||
|
|
g.Expect(a).To(Equal(b))
|
||
|
|
}
|