mirror of
https://github.com/ipfs/kubo.git
synced 2025-08-15 20:01:44 +08:00
f1ae13d721
1. Use a custom bitfield type instead of bigints. 2. Make iterating over a hamt *significantly* faster. License: MIT Signed-off-by: Steven Allen <steven@stebalien.com>
46 lines
831 B
Go
46 lines
831 B
Go
package hamt
|
|
|
|
import (
|
|
"testing"
|
|
)
|
|
|
|
func TestHashBitsEvenSizes(t *testing.T) {
|
|
buf := []byte{255, 127, 79, 45, 116, 99, 35, 17}
|
|
hb := hashBits{b: buf}
|
|
|
|
for _, v := range buf {
|
|
if hb.Next(8) != int(v) {
|
|
t.Fatal("got wrong numbers back")
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestHashBitsUneven(t *testing.T) {
|
|
buf := []byte{255, 127, 79, 45, 116, 99, 35, 17}
|
|
hb := hashBits{b: buf}
|
|
|
|
v := hb.Next(4)
|
|
if v != 15 {
|
|
t.Fatal("should have gotten 15: ", v)
|
|
}
|
|
|
|
v = hb.Next(4)
|
|
if v != 15 {
|
|
t.Fatal("should have gotten 15: ", v)
|
|
}
|
|
|
|
if v := hb.Next(3); v != 3 {
|
|
t.Fatalf("expected 3, but got %b", v)
|
|
}
|
|
if v := hb.Next(3); v != 7 {
|
|
t.Fatalf("expected 7, but got %b", v)
|
|
}
|
|
if v := hb.Next(3); v != 6 {
|
|
t.Fatalf("expected 6, but got %b", v)
|
|
}
|
|
|
|
if v := hb.Next(15); v != 20269 {
|
|
t.Fatalf("expected 20269, but got %b (%d)", v, v)
|
|
}
|
|
}
|