From 830a44a84a61b41b393e2105846f4f8ce07c55b5 Mon Sep 17 00:00:00 2001
From: chrislusf <chris.lu@gmail.com>
Date: Mon, 13 Apr 2015 23:39:11 -0700
Subject: [PATCH] Add compression test

---
 go/compress/compression_test.go | 45 +++++++++++++++++++++++++++++++++
 1 file changed, 45 insertions(+)
 create mode 100644 go/compress/compression_test.go

diff --git a/go/compress/compression_test.go b/go/compress/compression_test.go
new file mode 100644
index 000000000..83b7c0055
--- /dev/null
+++ b/go/compress/compression_test.go
@@ -0,0 +1,45 @@
+package compress
+
+import (
+	"math/rand"
+	"testing"
+)
+
+func TestSortedData(t *testing.T) {
+	data := make([]int32, 102400)
+	for i := 1; i < len(data); i++ {
+		data[i] = data[i-1] + rand.Int31n(15)
+	}
+	testCompressAndUncompress(t, data, "Sorted data")
+}
+
+func TestUnsortedData(t *testing.T) {
+	data := make([]int32, 102400)
+	for i := 0; i < len(data); i++ {
+		data[i] = rand.Int31n(255)
+	}
+	testCompressAndUncompress(t, data, "Unsorted data")
+}
+
+func testCompressAndUncompress(t *testing.T, data []int32, desc string) {
+
+	compressed_data, err := Compress32(data)
+	if err != nil {
+		t.Fatal("Compress error", err.Error())
+	}
+	uncompressed_data, err := Uncompress32(compressed_data, make([]int32, len(data)*2))
+	if err != nil {
+		t.Fatal("Compress error", err.Error())
+	}
+	if len(uncompressed_data) != len(data) {
+		t.Fatal("Len differs", len(data), len(uncompressed_data))
+	}
+	for i := 0; i < len(data); i++ {
+		if data[i] != uncompressed_data[i] {
+			t.Fatal("Data differs:", i, data[i], uncompressed_data[i])
+		}
+	}
+
+	println(desc, " Data length:", len(data), " => Compressed length:", len(compressed_data))
+
+}