use of edu.uci.ics.texera.dataflow.sink.tuple.TupleSink in project textdb by TextDB.
the class EmojiSentimentTest method test5.
@Test
public void test5() throws TexeraException {
TupleSourceOperator tupleSource = new TupleSourceOperator(Arrays.asList(EmojiSentimentTestConstants.NEGATIVE_TUPLE2), EmojiSentimentTestConstants.SENTIMENT_SCHEMA);
EmojiSentimentOperator sentiment = new EmojiSentimentOperator(new EmojiSentimentPredicate(NlpSentimentTestConstants.TEXT, "sentiment"));
TupleSink tupleSink = new TupleSink();
sentiment.setInputOperator(tupleSource);
tupleSink.setInputOperator(sentiment);
tupleSink.open();
List<Tuple> results = tupleSink.collectAllTuples();
tupleSink.close();
Tuple tuple = results.get(0);
Assert.assertEquals(tuple.getField("sentiment").getValue(), SentimentConstants.NEGATIVE);
}
use of edu.uci.ics.texera.dataflow.sink.tuple.TupleSink in project textdb by TextDB.
the class EmojiSentimentTest method test4.
@Test
public void test4() throws TexeraException {
TupleSourceOperator tupleSource = new TupleSourceOperator(Arrays.asList(EmojiSentimentTestConstants.POSITIVE_TUPLE2), EmojiSentimentTestConstants.SENTIMENT_SCHEMA);
EmojiSentimentOperator sentiment = new EmojiSentimentOperator(new EmojiSentimentPredicate(EmojiSentimentTestConstants.TEXT, "sentiment"));
TupleSink tupleSink = new TupleSink();
sentiment.setInputOperator(tupleSource);
tupleSink.setInputOperator(sentiment);
tupleSink.open();
List<Tuple> results = tupleSink.collectAllTuples();
tupleSink.close();
Tuple tuple = results.get(0);
Assert.assertEquals(tuple.getField("sentiment").getValue(), SentimentConstants.POSITIVE);
}
use of edu.uci.ics.texera.dataflow.sink.tuple.TupleSink in project textdb by TextDB.
the class NlpSentimentTest method test1.
/*
* Test sentiment with a positive sentence, result should be 3 (positive)
*/
@Test
public void test1() throws TexeraException {
TupleSourceOperator tupleSource = new TupleSourceOperator(Arrays.asList(NlpSentimentTestConstants.POSITIVE_TUPLE), NlpSentimentTestConstants.SENTIMENT_SCHEMA);
NlpSentimentOperator sentiment = new NlpSentimentOperator(new NlpSentimentPredicate(NlpSentimentTestConstants.TEXT, "sentiment"));
TupleSink tupleSink = new TupleSink();
sentiment.setInputOperator(tupleSource);
tupleSink.setInputOperator(sentiment);
tupleSink.open();
List<Tuple> results = tupleSink.collectAllTuples();
tupleSink.close();
Tuple tuple = results.get(0);
Assert.assertEquals(tuple.getField("sentiment").getValue(), SentimentConstants.POSITIVE);
}
use of edu.uci.ics.texera.dataflow.sink.tuple.TupleSink in project textdb by TextDB.
the class NlpSentimentTest method test3.
/*
* Test sentiment with a negative sentence, result should be 1 (negative)
*/
@Test
public void test3() throws TexeraException {
TupleSourceOperator tupleSource = new TupleSourceOperator(Arrays.asList(NlpSentimentTestConstants.NEGATIVE_TUPLE), NlpSentimentTestConstants.SENTIMENT_SCHEMA);
NlpSentimentOperator sentiment = new NlpSentimentOperator(new NlpSentimentPredicate(NlpSentimentTestConstants.TEXT, "sentiment"));
TupleSink tupleSink = new TupleSink();
sentiment.setInputOperator(tupleSource);
tupleSink.setInputOperator(sentiment);
tupleSink.open();
List<Tuple> results = tupleSink.collectAllTuples();
tupleSink.close();
Tuple tuple = results.get(0);
Assert.assertEquals(tuple.getField("sentiment").getValue(), SentimentConstants.NEGATIVE);
}
use of edu.uci.ics.texera.dataflow.sink.tuple.TupleSink in project textdb by TextDB.
the class NltkSentimentOperatorTest method test1.
/*
* Test sentiment test result should be positive.
*/
@Test
public void test1() throws TexeraException {
TupleSourceOperator tupleSource = new TupleSourceOperator(Arrays.asList(NltkSentimentTestConstants.POSITIVE_TUPLE), NlpSentimentTestConstants.SENTIMENT_SCHEMA);
NltkSentimentOperator nltkSentimentOperator = new NltkSentimentOperator(new NltkSentimentOperatorPredicate(NlpSentimentTestConstants.TEXT, "sentiment", BATCH_SIZE, MODEL_FILE_NAME));
TupleSink tupleSink = new TupleSink();
nltkSentimentOperator.setInputOperator(tupleSource);
tupleSink.setInputOperator(nltkSentimentOperator);
tupleSink.open();
List<Tuple> results = tupleSink.collectAllTuples();
tupleSink.close();
Tuple tuple = results.get(0);
Assert.assertEquals(tuple.getField("sentiment").getValue(), SentimentConstants.POSITIVE);
}
Aggregations