This commit is contained in:
parent
d077d894fb
commit
dbc0a96156
21
src/test/java/org/studiorailgun/AnswerTests.java
Normal file
21
src/test/java/org/studiorailgun/AnswerTests.java
Normal file
@ -0,0 +1,21 @@
|
||||
package org.studiorailgun;
|
||||
|
||||
import static org.junit.Assert.*;
|
||||
|
||||
import org.junit.Test;
|
||||
import org.studiorailgun.conversation.evaluators.transfer.AnswerSynthesis;
|
||||
import org.studiorailgun.conversation.tracking.Quote;
|
||||
|
||||
/**
|
||||
* Answer queries
|
||||
*/
|
||||
public class AnswerTests {
|
||||
|
||||
@Test
|
||||
public void testAnswerQuery(){
|
||||
Globals.init("./data/webs/test/web.json");
|
||||
Quote result = AnswerSynthesis.evaluate(Globals.conversation, new Quote("What color is your hat?"));
|
||||
assertEquals(result.getRaw().contains("Blue"), true);
|
||||
}
|
||||
|
||||
}
|
||||
@ -4,6 +4,8 @@ import static org.junit.jupiter.api.Assertions.*;
|
||||
|
||||
import org.junit.jupiter.api.Test;
|
||||
import org.studiorailgun.conversation.ConvAI;
|
||||
import org.studiorailgun.conversation.evaluators.query.QueryData;
|
||||
import org.studiorailgun.conversation.evaluators.query.QueryEval;
|
||||
import org.studiorailgun.conversation.tracking.Quote;
|
||||
|
||||
/**
|
||||
@ -21,4 +23,13 @@ public class QueryTests {
|
||||
assertEquals(responseContainsBlue, true);
|
||||
}
|
||||
|
||||
|
||||
@Test
|
||||
public void testQueryEval(){
|
||||
Globals.init("./data/webs/test/web.json");
|
||||
QueryEval.evaluate(Globals.conversation, new Quote("What color is your hat?"));
|
||||
QueryData queryData = Globals.conversation.getGoalData().getQueryData();
|
||||
assertEquals(queryData.getRecentQueries().size(), 1);
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
Loading…
Reference in New Issue
Block a user