closestmatch_test.go

package closestmatch

import (
	"fmt"
	"io/ioutil"
	"strings"
	"testing"

	"github.com/schollz/closestmatch/test"
)

func BenchmarkNew(b *testing.B) {
	for i := 0; i < b.N; i++ {
		New(test.WordsToTest, []int{3})
	}
}

func BenchmarkSplitOne(b *testing.B) {
	cm := New(test.WordsToTest, []int{3})
	searchWord := test.SearchWords[0]
	b.ResetTimer()
	for i := 0; i < b.N; i++ {
		cm.splitWord(searchWord)
	}
}

func BenchmarkClosestOne(b *testing.B) {
	bText, _ := ioutil.ReadFile("test/books.list")
	wordsToTest := strings.Split(strings.ToLower(string(bText)), "\n")
	cm := New(wordsToTest, []int{3})
	searchWord := test.SearchWords[0]
	b.ResetTimer()
	for i := 0; i < b.N; i++ {
		cm.Closest(searchWord)
	}
}

func BenchmarkClosest3(b *testing.B) {
	bText, _ := ioutil.ReadFile("test/books.list")
	wordsToTest := strings.Split(strings.ToLower(string(bText)), "\n")
	cm := New(wordsToTest, []int{3})
	searchWord := test.SearchWords[0]
	b.ResetTimer()
	for i := 0; i < b.N; i++ {
		cm.ClosestN(searchWord, 3)
	}
}

func BenchmarkClosest30(b *testing.B) {
	bText, _ := ioutil.ReadFile("test/books.list")
	wordsToTest := strings.Split(strings.ToLower(string(bText)), "\n")
	cm := New(wordsToTest, []int{3})
	searchWord := test.SearchWords[0]
	b.ResetTimer()
	for i := 0; i < b.N; i++ {
		cm.ClosestN(searchWord, 30)
	}
}

func BenchmarkFileLoad(b *testing.B) {
	bText, _ := ioutil.ReadFile("test/books.list")
	wordsToTest := strings.Split(strings.ToLower(string(bText)), "\n")
	cm := New(wordsToTest, []int{3, 4})
	cm.Save("test/books.list.cm.gz")
	b.ResetTimer()
	for i := 0; i < b.N; i++ {
		Load("test/books.list.cm.gz")
	}
}

func BenchmarkFileSave(b *testing.B) {
	bText, _ := ioutil.ReadFile("test/books.list")
	wordsToTest := strings.Split(strings.ToLower(string(bText)), "\n")
	cm := New(wordsToTest, []int{3, 4})
	b.ResetTimer()
	for i := 0; i < b.N; i++ {
		cm.Save("test/books.list.cm.gz")
	}
}

func ExampleMatchingSmall() {
	cm := New([]string{"love", "loving", "cat", "kit", "cats"}, []int{4})
	fmt.Println(cm.splitWord("love"))
	fmt.Println(cm.splitWord("kit"))
	fmt.Println(cm.Closest("kit"))
	// Output:
	// map[love:{}]
	// map[kit:{}]
	// kit

}

func ExampleMatchingSimple() {
	cm := New(test.WordsToTest, []int{3})
	for _, searchWord := range test.SearchWords {
		fmt.Printf("'%s' matched '%s'\n", searchWord, cm.Closest(searchWord))
	}
	// Output:
	// 'cervantes don quixote' matched 'don quixote by miguel de cervantes saavedra'
	// 'mysterious afur at styles by christie' matched 'the mysterious affair at styles by agatha christie'
	// 'hard times by charles dickens' matched 'hard times by charles dickens'
	// 'complete william shakespeare' matched 'the complete works of william shakespeare by william shakespeare'
	// 'war by hg wells' matched 'the war of the worlds by h. g. wells'

}

func ExampleMatchingN() {
	cm := New(test.WordsToTest, []int{4})
	fmt.Println(cm.ClosestN("war h.g. wells", 3))
	// Output:
	// [the war of the worlds by h. g. wells the time machine by h. g. wells war and peace by graf leo tolstoy]
}

func ExampleMatchingBigList() {
	bText, _ := ioutil.ReadFile("test/books.list")
	wordsToTest := strings.Split(strings.ToLower(string(bText)), "\n")
	cm := New(wordsToTest, []int{3})
	searchWord := "island of a thod mirrors"
	fmt.Println(cm.Closest(searchWord))
	// Output:
	// island of a thousand mirrors by nayomi munaweera
}

func ExampleMatchingCatcher() {
	bText, _ := ioutil.ReadFile("test/catcher.txt")
	wordsToTest := strings.Split(strings.ToLower(string(bText)), "\n")
	cm := New(wordsToTest, []int{5})
	searchWord := "catcher in the rye by jd salinger"
	for i, match := range cm.ClosestN(searchWord, 3) {
		if i == 2 {
			fmt.Println(match)
		}
	}
	// Output:
	// the catcher in the rye by j.d. salinger
}

func ExampleMatchingPotter() {
	bText, _ := ioutil.ReadFile("test/potter.txt")
	wordsToTest := strings.Split(strings.ToLower(string(bText)), "\n")
	cm := New(wordsToTest, []int{5})
	searchWord := "harry potter and the half blood prince by j.k. rowling"
	for i, match := range cm.ClosestN(searchWord, 3) {
		if i == 1 {
			fmt.Println(match)
		}
	}
	// Output:
	//  harry potter and the order of the phoenix (harry potter, #5, part 1) by j.k. rowling
}

func TestAccuracyBookWords(t *testing.T) {
	bText, _ := ioutil.ReadFile("test/books.list")
	wordsToTest := strings.Split(strings.ToLower(string(bText)), "\n")
	cm := New(wordsToTest, []int{4, 5})
	accuracy := cm.AccuracyMutatingWords()
	fmt.Printf("Accuracy with mutating words in book list:\t%2.1f%%\n", accuracy)
}

func TestAccuracyBookLetters(t *testing.T) {
	bText, _ := ioutil.ReadFile("test/books.list")
	wordsToTest := strings.Split(strings.ToLower(string(bText)), "\n")
	cm := New(wordsToTest, []int{5})
	accuracy := cm.AccuracyMutatingLetters()
	fmt.Printf("Accuracy with mutating letters in book list:\t%2.1f%%\n", accuracy)
}

func TestAccuracyDictionaryLetters(t *testing.T) {
	bText, _ := ioutil.ReadFile("test/popular.txt")
	wordsToTest := strings.Split(strings.ToLower(string(bText)), "\n")
	cm := New(wordsToTest, []int{2, 3, 4})
	accuracy := cm.AccuracyMutatingWords()
	fmt.Printf("Accuracy with mutating letters in dictionary:\t%2.1f%%\n", accuracy)
}

func TestSaveLoad(t *testing.T) {
	bText, _ := ioutil.ReadFile("test/books.list")
	wordsToTest := strings.Split(strings.ToLower(string(bText)), "\n")
	type TestStruct struct {
		cm *ClosestMatch
	}
	tst := new(TestStruct)
	tst.cm = New(wordsToTest, []int{5})
	err := tst.cm.Save("test.gob")
	if err != nil {
		t.Error(err)
	}

	tst2 := new(TestStruct)
	tst2.cm, err = Load("test.gob")
	if err != nil {
		t.Error(err)
	}
	answer2 := tst2.cm.Closest("war of the worlds by hg wells")
	answer1 := tst.cm.Closest("war of the worlds by hg wells")
	if answer1 != answer2 {
		t.Errorf("Differing answers: '%s' '%s'", answer1, answer2)
	}
}