mirror of
https://github.com/taigrr/arc
synced 2025-01-18 04:33:13 -08:00
90 lines
1.6 KiB
Go
90 lines
1.6 KiB
Go
package sss
|
|
|
|
import (
|
|
"bytes"
|
|
"testing"
|
|
)
|
|
|
|
var (
|
|
p = []byte{1, 0, 2, 3}
|
|
p2 = []byte{70, 32, 6}
|
|
)
|
|
|
|
func TestDegree(t *testing.T) {
|
|
if v, want := degree(p), 3; v != want {
|
|
t.Errorf("Was %v, but expected %v", v, want)
|
|
}
|
|
}
|
|
|
|
func TestEval(t *testing.T) {
|
|
if v, want := eval(p, 2), byte(17); v != want {
|
|
t.Errorf("Was %v, but expected %v", v, want)
|
|
}
|
|
}
|
|
|
|
func TestGenerate(t *testing.T) {
|
|
b := []byte{1, 2, 3}
|
|
|
|
expected := []byte{10, 1, 2, 3}
|
|
actual, err := generate(3, 10, bytes.NewReader(b))
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
if !bytes.Equal(actual, expected) {
|
|
t.Errorf("Was %v, but expected %v", actual, expected)
|
|
}
|
|
}
|
|
|
|
func TestGenerateEOF(t *testing.T) {
|
|
b := []byte{1}
|
|
|
|
p, err := generate(3, 10, bytes.NewReader(b))
|
|
if p != nil {
|
|
t.Errorf("Was %v, but expected an error", p)
|
|
}
|
|
|
|
if err == nil {
|
|
t.Error("No error returned")
|
|
}
|
|
}
|
|
|
|
func TestGeneratePolyEOFFullSize(t *testing.T) {
|
|
b := []byte{1, 2, 0, 0, 0, 0}
|
|
|
|
p, err := generate(3, 10, bytes.NewReader(b))
|
|
if p != nil {
|
|
t.Errorf("Was %v, but xpected an error", p)
|
|
}
|
|
|
|
if err == nil {
|
|
t.Error("No error returned")
|
|
}
|
|
}
|
|
|
|
func TestGenerateFullSize(t *testing.T) {
|
|
b := []byte{1, 2, 0, 4}
|
|
|
|
expected := []byte{10, 1, 2, 4}
|
|
actual, err := generate(3, 10, bytes.NewReader(b))
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
if !bytes.Equal(actual, expected) {
|
|
t.Errorf("Was %v but expected %v", actual, expected)
|
|
}
|
|
}
|
|
|
|
func TestInterpolate(t *testing.T) {
|
|
in := []pair{
|
|
pair{x: 1, y: 1},
|
|
pair{x: 2, y: 2},
|
|
pair{x: 3, y: 3},
|
|
}
|
|
|
|
if v, want := interpolate(in, 0), byte(0); v != want {
|
|
t.Errorf("Was %v, but expected %v", v, want)
|
|
}
|
|
}
|