multiprobe_test.go

package lsh

import (
	"strconv"
	"testing"
)

func Test_NewMultiprobeLsh(t *testing.T) {
	lsh := NewMultiprobeLsh(100, 5, 5, 5.0, 64)
	if len(lsh.tables) != 5 {
		t.Error("Lsh init fail")
	}
	t.Logf("Scores %v", lsh.scores)
	t.Logf("Perturbation sets: %v", lsh.perturbSets)
	for i, v := range lsh.perturbSets {
		t.Logf("Set: %d, Score: %f, Set contents: %v", i, lsh.getScore(&v), v)
	}
	for i, perSet := range lsh.perturbVecs {
		for j, perTable := range perSet {
			t.Logf("Set: %d, Table: %d, Vec: %v", i, j, perTable)
		}
	}

}

func Test_MultiprobeLshQueryKnn(t *testing.T) {
	lsh := NewMultiprobeLsh(100, 5, 5, 5.0, 10)
	points := randomPoints(10, 100, 32.0)
	insertedKeys := make([]string, 10)
	for i, p := range points {
		lsh.Insert(p, strconv.Itoa(i))
		insertedKeys[i] = strconv.Itoa(i)
	}
	// Use the inserted points as queries, and
	// verify that we can get back each query itself
	for i, key := range insertedKeys {
		found := false
		for _, foundKey := range lsh.Query(points[i]) {
			if foundKey == key {
				found = true
			}
		}
		if !found {
			t.Error("Query fail")
		}
	}
}