Function ModelTest

coder/arith/models_test.go:9–31

Tests a model provided by the constructor

(t *testing.T, N int, model func() Model)

// Tests a model provided by the constructor
func ModelTest(t *testing.T, N int, model func() Model) {
	encm, decm := model(), model()
	mask := uint(1<<encm.NBits() - 1)

	bits := make([]uint, N)
	for i := range bits {
		bits[i] = uint(rand.Int()) & mask
	}

	enc := NewEncoder()
	for _, bit := range bits {
		encm.Encode(enc, bit)
	}
	enc.Close()

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

func TestShiftModel(t *testing.T) {
	ModelTest(t, 3, func() Model { return &Shift{Prob(0.5), 3} })