fix PaletteContainer bug

level/palette_test.go

@@ -1,12 +1,14 @@
 package level
 
 import (
+	"math/rand"
 	"testing"
+
+	"github.com/Tnze/go-mc/level/block"
 )
 
-func TestPaletteResize(t *testing.T) {
-	container := NewStatesPaletteContainer(16*16*16, 0)
-
+func TestPaletteContainer_seq(t *testing.T) {
+	container := NewStatesPaletteContainer(4096, 0)
 	for i := 0; i < 4096; i++ {
 		container.Set(i, BlocksState(i))
 	}
@@ -16,3 +18,41 @@ func TestPaletteResize(t *testing.T) {
 		}
 	}
 }
+
+func TestPaletteContainer_rand(t *testing.T) {
+	data := make([]BlocksState, 4096)
+	for i := range data {
+		data[i] = BlocksState(rand.Intn(1 << block.BitsPerBlock))
+	}
+	container := NewStatesPaletteContainer(4096, 0)
+	for i, v := range data {
+		container.Set(i, v)
+	}
+	for i, v := range data {
+		if v2 := container.Get(i); v != v2 {
+			t.Errorf("value not match: got %v, except: %v", v2, v)
+		}
+	}
+}
+
+func BenchmarkPaletteContainer(b *testing.B) {
+	data := make([]BlocksState, 4096)
+	for i := range data {
+		data[i] = BlocksState(rand.Intn(1 << block.BitsPerBlock))
+	}
+	rand.Shuffle(len(data), func(i, j int) { data[i], data[j] = data[j], data[i] })
+	container := NewStatesPaletteContainer(4096, 0)
+	b.ResetTimer()
+	b.Run("Set", func(b *testing.B) {
+		for i := 0; i < b.N; i++ {
+			index := i % 4096
+			container.Set(index, data[index])
+		}
+	})
+	b.Run("Get", func(b *testing.B) {
+		for i := 0; i < b.N; i++ {
+			index := i % 4096
+			container.Get(index)
+		}
+	})
+}