@@ -22,16 +22,19 @@ func TestCoding(t *testing.T) { n := New(32) for i := uint64(0); i < 4096; i++ { - enc, _ := n.Code(i) - //fmt.Printf("%d - %b, %d\n", i, enc, len) - dec, _ := n.Decode(enc) + enc, encLen := n.Code(i) + dec, decLen := n.Decode(enc) + if i != dec { t.Errorf("Unexpected value for %d - enc is %b, dec is %d\n", i, enc, dec) } + if encLen != decLen { + t.Errorf("Unexpected difference between encoded and decoded lengths.", encLen, decLen) + } } } func TestWriter(t *testing.T) { var buf bytes.Buffer