func TestEncode(t *testing.T) {
var tests = []struct {
numECBytes int
data string
rsCode string
}{
{
5,
"01000000 00011000 10101100 11000011 00000000",
"01000000 00011000 10101100 11000011 00000000 10000110 00001101 00100010 10101110 00110000",
},
{
10,
"00010000 00100000 00001100 01010110 01100001 10000000 11101100 00010001 11101100 00010001 11101100 00010001 11101100 00010001 11101100 00010001",
"00010000 00100000 00001100 01010110 01100001 10000000 11101100 00010001 11101100 00010001 11101100 00010001 11101100 00010001 11101100 00010001 10100101 00100100 11010100 11000001 11101101 00110110 11000111 10000111 00101100 01010101",
},
}
for _, test := range tests {
data := bitset.NewFromBase2String(test.data)
rsCode := bitset.NewFromBase2String(test.rsCode)
result := Encode(data, test.numECBytes)
if !rsCode.Equals(result) {
t.Errorf("data=%s, numECBytes=%d, encoded=%s, want %s",
data.String(),
test.numECBytes,
result.String(),
rsCode)
}
}
}
func TestSingleModeEncodings(t *testing.T) {
tests := []struct {
dataEncoderType dataEncoderType
dataMode dataMode
data string
expected *bitset.Bitset
}{
{
dataEncoderType1To9,
dataModeNumeric,
"01234567",
bitset.NewFromBase2String("0001 0000001000 0000001100 0101011001 1000011"),
},
{
dataEncoderType1To9,
dataModeAlphanumeric,
"AC-42",
bitset.NewFromBase2String("0010 000000101 00111001110 11100111001 000010"),
},
{
dataEncoderType1To9,
dataModeByte,
"123",
bitset.NewFromBase2String("0100 00000011 00110001 00110010 00110011"),
},
{
dataEncoderType10To26,
dataModeByte,
"123",
bitset.NewFromBase2String("0100 00000000 00000011 00110001 00110010 00110011"),
},
{
dataEncoderType27To40,
dataModeByte,
"123",
bitset.NewFromBase2String("0100 00000000 00000011 00110001 00110010 00110011"),
},
}
for _, test := range tests {
encoder := newDataEncoder(test.dataEncoderType)
encoded := bitset.New()
encoder.encodeDataRaw([]byte(test.data), test.dataMode, encoded)
if !test.expected.Equals(encoded) {
t.Errorf("For %s got %s, expected %s", test.data, encoded.String(),
test.expected.String())
}
}
}