func TestFindBasisAndSort(t *testing.T) {
constr := fastTestConstruction()
S := GenerateS(constr, 0, 0, 0)
basis := FindBasisAndSort(S)
// Test that each vector is correct in place.
for i, perm := range S {
vect := [256]byte{}
copy(vect[:], table.SerializeByte(FunctionFromBasis(i, basis)))
if vect != perm {
t.Fatalf("FindBasisAndSort, position #%v is wrong!\n%v\n%v", i, vect, perm)
}
}
// Test composition of two functions.
x := table.SerializeByte(table.ComposedBytes{
table.ParsedByte(S[39][:]),
table.ParsedByte(S[120][:]),
})
y := [256]byte{}
copy(y[:], x)
if y != S[39^120] {
t.Fatalf("Function composition was wrong!\n%v\n%v", x, S[39^120])
}
}
// FindBasisAndSort finds 8 elements of S that act as a basis for S and build isomorphism psi.
func FindBasisAndSort(S [][256]byte) (basis []table.Byte) {
for len(basis) < 8 { // Until we have a full basis.
basis = append(basis, table.ParsedByte(S[1<<uint(len(basis))][:])) // Add the first independent vector to the basis.
// Move all (now) dependent vectors from S into their correct position.
for i := 1 << uint(len(basis)-1); i < 1<<uint(len(basis)); i++ {
vect := [256]byte{}
copy(vect[:], table.SerializeByte(FunctionFromBasis(i, basis)))
// Move it to the correct position in S.
for j := i; j < len(S); j++ {
if vect == S[j] {
S[i], S[j] = S[j], S[i]
break
}
}
}
}
return
}
