Fix a major bug, forgot to add intentsPath to olivia_test

olivia-ai · Mar 25, 2020 · 1ea6cd4 · 1ea6cd4
1 parent 6592023
commit 1ea6cd4
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/olivia_test.go b/olivia_test.go
@@ -11,7 +11,8 @@ import (
 )
 
 func TestCalculate(t *testing.T) {
-	model := training.CreateNeuralNetwork()
+	intentsPath := "res/intents.json"
+	model := training.CreateNeuralNetwork(intentsPath)
 	cache := gocache.New(5*time.Minute, 5*time.Minute)
 
 	sentences := map[string]string{
@@ -32,7 +33,7 @@ func TestCalculate(t *testing.T) {
 	}
 
 	for sentence, tag := range sentences {
-		responseTag, _ := analysis.NewSentence(sentence).Calculate(*cache, model, "1")
+		responseTag, _ := analysis.NewSentence(sentence).Calculate(*cache, model, intentsPath, "1")
 
 		if !regexp.MustCompile(tag).MatchString(responseTag) {
 			t.Errorf("Expected \"%s\" tag for \"%s\", found \"%s\"", tag, sentence, responseTag)