1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
|
package ff
import (
"math/big"
"math/rand"
"reflect"
"testing"
)
func TestElementToBigInt(t *testing.T) {
type args struct {
a Element
}
tests := []struct {
name string
args args
wantZ *big.Int
}{
{"0", args{Element{0, 0, 0, 0}}, big.NewInt(0)},
{"1", args{Element{1, 0, 0, 0}}, big.NewInt(1)},
{"1", args{Element{1, 0, 0, 0}}, big.NewInt(1)},
{"2^64", args{Element{0, 1, 0, 0}}, new(big.Int).Lsh(big.NewInt(1), 64)},
{"p", args{p}, pBI},
}
for _, tt := range tests {
t.Run(tt.name, func(t *testing.T) {
if gotZ := ElementToBigInt(tt.args.a); !reflect.DeepEqual(gotZ, tt.wantZ) {
t.Errorf("ElementToBigInt() = %v, want %v", gotZ, tt.wantZ)
}
})
}
}
func TestBigIntToElement(t *testing.T) {
type args struct {
z *big.Int
}
tests := []struct {
name string
args args
wantA Element
}{
{"0", args{big.NewInt(0)}, Element{0, 0, 0, 0}},
{"1", args{big.NewInt(1)}, Element{1, 0, 0, 0}},
{"2^64", args{new(big.Int).Lsh(big.NewInt(1), 64)}, Element{0, 1, 0, 0}},
{"p", args{pBI}, p},
}
for _, tt := range tests {
t.Run(tt.name, func(t *testing.T) {
if gotA := BigIntToElement(tt.args.z); !reflect.DeepEqual(gotA, tt.wantA) {
t.Errorf("BigIntToElement() = %v, want %v", gotA, tt.wantA)
}
})
}
}
func Test_BigIntToElement_ElementToBigInt_random(t *testing.T) {
for i := 0; i < 10; i++ {
a := randomElement(i)
n := ElementToBigInt(a)
aa := BigIntToElement(n)
if a != aa {
t.Errorf("/%d BigIntToElement(ElementToBigInt(%v)) = %v, want %v", i, a, aa, a)
t.FailNow()
}
nn := ElementToBigInt(aa)
if n.Cmp(nn) != 0 {
t.Errorf("/%d ElementToBigInt(BigIntToElement(%v)) = %v, want %v", i, n, nn, n)
t.FailNow()
}
}
}
func randomElement(seed int) Element {
r := rand.NewSource(int64(seed))
return Element{uint64(r.Int63()), uint64(r.Int63()), uint64(r.Int63()), uint64(r.Int63())}
}
|
