Function TestByteModel

coder/arith/models_test.go:62–86

(t *testing.T)

}

func TestByteModel(t *testing.T) {
	B := func() Model { return &Shift{Prob(0.5), 5} }
	T := func() Model { return NewTree(8, B) }

	encm, decm := T(), T()

	data := make([]byte, 25192)
	for i := range data {
		data[i] = byte(rand.Int())
	}

	enc := NewEncoder()
	for _, b := range data {
		encm.Encode(enc, uint(b))
	}
	enc.Close()

	dec := NewDecoder(enc.Bytes())
	for i, v := range data {
		x := decm.Decode(dec)
		if byte(x) != v {
			t.Fatalf("fail %v: got %v exp %v", i, x, v)
		}
	}
}

func TestShiftZeros(t *testing.T) {
	model := func() Model { return &Shift{MaxP / 1000, 7} }