added test for training model

Author: ikwattro

src/main/java/com/graphaware/nlp/processor/opennlp/OpenNLPPipeline.java

@@ -365,7 +365,7 @@ else if (alg.toLowerCase().equals("sentiment")) {
             sentModel.train();
             result = sentModel.validate();
             String[] dirPathSplit = fileTrain.split(File.separator);
-            String fileOutToUse = null;
+            String fileOutToUse;
             if (dirPathSplit.length > 2) {
                 StringBuilder sb = new StringBuilder("");
                 for (int i = 0; i < dirPathSplit.length -2; ++i) {

src/test/java/com/graphaware/nlp/processor/opennlp/model/CustomSentimentModelIntegrationTest.java

@@ -3,6 +3,8 @@
 import com.graphaware.nlp.NLPIntegrationTest;
 import org.junit.Test;
 
+import java.util.Map;
+
 import static org.junit.Assert.*;
 
 public class CustomSentimentModelIntegrationTest extends NLPIntegrationTest {
@@ -14,6 +16,23 @@ public void testTrainCustomModelWithProcedure() {
         executeInTransaction(q, (result -> {
             assertTrue(result.hasNext());
         }));
+
+        String addPipelineQuery = "CALL ga.nlp.processor.addPipeline({textProcessor: 'com.graphaware.nlp.processor.opennlp.OpenNLPTextProcessor', name: 'customSentiment', processingSteps: {tokenize: true, ner: true, sentiment: true, dependency: false, customSentiment: \"component-en\"}})";
+        executeInTransaction(addPipelineQuery, emptyConsumer());
+
+        String insertQ = "CREATE (tweet:Tweet) SET tweet.text = \"African American unemployment is the lowest ever recorded in our country. The Hispanic unemployment rate dropped a full point in the last year and is close to the lowest in recorded history. Dems did nothing for you but get your vote!\"\n" +
+                "WITH tweet\n" +
+                "CALL ga.nlp.annotate({text:tweet.text, id:id(tweet), pipeline:\"customSentiment\", checkLanguage:false})\n" +
+                "YIELD result\n" +
+                "MERGE (tweet)-[:HAS_ANNOTATED_TEXT]->(result)";
+        executeInTransaction(insertQ, emptyConsumer());
+        executeInTransaction("MATCH (n:Sentence) RETURN ANY(x IN labels(n) WHERE x IN ['Positive','Very Positive','Neutral']) AS hasSentiment", (result -> {
+            assertTrue(result.hasNext());
+            while (result.hasNext()) {
+                Map<String, Object> record = result.next();
+                assertTrue((boolean) record.get("hasSentiment"));
+            }
+        }));
     }
 
 }