func TestCorrectness() { fmt.Printf("\n\nCorrectness test") // Test behavior for ii := 1; ii < 20; ii++ { fmt.Printf("\nTest %v", ii) var values []byte rand.Seed(time.Now().UTC().UnixNano()) if ii == 3 { values = []byte{0, 0, 32, 15, -4 & 0xFF, 16, 0, 16, 0, 7, -1 & 0xFF, -4 & 0xFF, -32 & 0xFF, 0, 31, -1 & 0xFF} } else if ii == 2 { values = []byte{0x3d, 0x4d, 0x54, 0x47, 0x5a, 0x36, 0x39, 0x26, 0x72, 0x6f, 0x6c, 0x65, 0x3d, 0x70, 0x72, 0x65} } else if ii == 4 { values = []byte{65, 71, 74, 66, 76, 65, 69, 77, 74, 79, 68, 75, 73, 72, 77, 68, 78, 65, 79, 79, 78, 66, 77, 71, 64, 70, 74, 77, 64, 67, 71, 64} } else if ii == 1 { values = make([]byte, 32) for i := range values { values[i] = byte(2) // all identical } } else if ii == 5 { values = make([]byte, 32) for i := range values { values[i] = byte(2 + (i & 1)) // 2 symbols } } else { values = make([]byte, 32) for i := range values { values[i] = byte(64 + 3*ii + rand.Intn(ii+1)) } } fmt.Printf("\nOriginal: \n") for i := range values { fmt.Printf("%d ", values[i]) } fmt.Printf("\nEncoded: \n") buffer := make([]byte, 16384) oFile, _ := util.NewByteArrayOutputStream(buffer, true) defer oFile.Close() obs, _ := bitstream.NewDefaultOutputBitStream(oFile, 16384) dbgbs, _ := bitstream.NewDebugOutputBitStream(obs, os.Stdout) dbgbs.ShowByte(true) dbgbs.Mark(true) hc, _ := entropy.NewHuffmanEncoder(dbgbs) if _, err := hc.Encode(values); err != nil { fmt.Printf("Error during encoding: %s", err) os.Exit(1) } hc.Dispose() dbgbs.Close() fmt.Printf("\nDecoded: \n") iFile, _ := util.NewByteArrayInputStream(buffer, true) defer iFile.Close() ibs, _ := bitstream.NewDefaultInputBitStream(iFile, 16384) dbgbs2, _ := bitstream.NewDebugInputBitStream(ibs, os.Stdout) //dbgbs2.ShowByte(true) dbgbs2.Mark(true) hd, _ := entropy.NewHuffmanDecoder(dbgbs2) ok := true values2 := make([]byte, len(values)) if _, err := hd.Decode(values2); err != nil { fmt.Printf("Error during decoding: %s", err) os.Exit(1) } println() for i := range values2 { fmt.Printf("%v ", values2[i]) if values[i] != values2[i] { ok = false } } if ok == true { fmt.Printf("\nIdentical") } else { fmt.Printf("\n! *** Different *** !") os.Exit(1) } hd.Dispose() println() } }
func TestSpeed() { fmt.Printf("\n\nSpeed test\n") repeats := []int{3, 1, 4, 1, 5, 9, 2, 6, 5, 3, 5, 8, 9, 7, 9, 3} for jj := 0; jj < 3; jj++ { fmt.Printf("Test %v\n", jj+1) delta1 := int64(0) delta2 := int64(0) iter := 4000 size := 50000 buffer := make([]byte, size*2) values1 := make([]byte, size) values2 := make([]byte, size) for ii := 0; ii < iter; ii++ { idx := jj for i := 0; i < len(values1); i++ { i0 := i length := repeats[idx] idx = (idx + 1) & 0x0F if i0+length >= len(values1) { length = 1 } b := byte(rand.Intn(256)) for j := i0; j < i0+length; j++ { values1[j] = b i++ } } oFile, _ := util.NewByteArrayOutputStream(buffer, false) defer oFile.Close() obs, _ := bitstream.NewDefaultOutputBitStream(oFile, uint(size)) rc, _ := entropy.NewHuffmanEncoder(obs) // Encode before := time.Now() if _, err := rc.Encode(values1); err != nil { fmt.Printf("An error occured during encoding: %v\n", err) os.Exit(1) } rc.Dispose() if _, err := obs.Close(); err != nil { fmt.Printf("Error during close: %v\n", err) os.Exit(1) } after := time.Now() delta1 += after.Sub(before).Nanoseconds() } for ii := 0; ii < iter; ii++ { iFile, _ := util.NewByteArrayInputStream(buffer, false) defer iFile.Close() ibs, _ := bitstream.NewDefaultInputBitStream(iFile, uint(size)) rd, _ := entropy.NewHuffmanDecoder(ibs) // Decode before := time.Now() if _, err := rd.Decode(values2); err != nil { fmt.Printf("An error occured during decoding: %v\n", err) os.Exit(1) } rd.Dispose() if _, err := ibs.Close(); err != nil { fmt.Printf("Error during close: %v\n", err) os.Exit(1) } after := time.Now() delta2 += after.Sub(before).Nanoseconds() } fmt.Printf("Encode [ms] : %d\n", delta1/1000000) fmt.Printf("Throughput [KB/s]: %d\n", (int64(iter*size))*1000000/delta1*1000/1024) fmt.Printf("Decode [ms] : %d\n", delta2/1000000) fmt.Printf("Throughput [KB/s]: %d\n", (int64(iter*size))*1000000/delta2*1000/1024) } }