use of edu.uci.ics.textdb.exp.source.scan.ScanSourcePredicate in project textdb by TextDB.
the class JoinTestHelper method getRegexMatcher.
public static RegexMatcher getRegexMatcher(String tableName, String query, String attrName) {
try {
ScanBasedSourceOperator scanBasedSourceOperator = new ScanBasedSourceOperator(new ScanSourcePredicate(tableName));
RegexMatcher regexMatcher = new RegexMatcher(new RegexPredicate(query, Arrays.asList(attrName), SchemaConstants.SPAN_LIST));
regexMatcher.setInputOperator(scanBasedSourceOperator);
return regexMatcher;
} catch (DataFlowException e) {
e.printStackTrace();
return null;
}
}
use of edu.uci.ics.textdb.exp.source.scan.ScanSourcePredicate in project textdb by TextDB.
the class ProjectionOperatorTest method testProjection2.
@Test
public void testProjection2() throws Exception {
List<String> projectionFields = Arrays.asList(TestConstants.FIRST_NAME, TestConstants.DESCRIPTION);
Schema projectionSchema = new Schema(TestConstants.FIRST_NAME_ATTR, TestConstants.DESCRIPTION_ATTR);
IField[] fields1 = { new StringField("bruce"), new TextField("Tall Angry") };
IField[] fields2 = { new StringField("tom hanks"), new TextField("Short Brown") };
IField[] fields3 = { new StringField("brad lie angelina"), new TextField("White Angry") };
IField[] fields4 = { new StringField("george lin lin"), new TextField("Lin Clooney is Short and lin clooney is Angry") };
IField[] fields5 = { new StringField("christian john wayne"), new TextField("Tall Fair") };
IField[] fields6 = { new StringField("Mary brown"), new TextField("Short angry") };
Tuple tuple1 = new Tuple(projectionSchema, fields1);
Tuple tuple2 = new Tuple(projectionSchema, fields2);
Tuple tuple3 = new Tuple(projectionSchema, fields3);
Tuple tuple4 = new Tuple(projectionSchema, fields4);
Tuple tuple5 = new Tuple(projectionSchema, fields5);
Tuple tuple6 = new Tuple(projectionSchema, fields6);
List<Tuple> expectedResults = Arrays.asList(tuple1, tuple2, tuple3, tuple4, tuple5, tuple6);
List<Tuple> returnedResults = getProjectionResults(new ScanBasedSourceOperator(new ScanSourcePredicate(PEOPLE_TABLE)), projectionFields);
Assert.assertTrue(TestUtils.equals(expectedResults, returnedResults));
}
use of edu.uci.ics.textdb.exp.source.scan.ScanSourcePredicate in project textdb by TextDB.
the class RegexMatcherTestHelper method getScanSourceResults.
public static List<Tuple> getScanSourceResults(String tableName, String regex, List<String> attributeNames, int limit, int offset) throws TextDBException {
ScanBasedSourceOperator scanSource = new ScanBasedSourceOperator(new ScanSourcePredicate(tableName));
RegexPredicate regexPredicate = new RegexPredicate(regex, attributeNames, RESULTS);
RegexMatcher regexMatcher = new RegexMatcher(regexPredicate);
regexMatcher.setLimit(limit);
regexMatcher.setOffset(offset);
regexMatcher.setInputOperator(scanSource);
Tuple tuple;
List<Tuple> results = new ArrayList<>();
regexMatcher.open();
while ((tuple = regexMatcher.getNextTuple()) != null) {
results.add(tuple);
}
regexMatcher.close();
return results;
}
use of edu.uci.ics.textdb.exp.source.scan.ScanSourcePredicate in project textdb by TextDB.
the class RegexSplitOperatorTest method computeRegexSplitResults.
public static List<Tuple> computeRegexSplitResults(String tableName, String splitAttrName, String splitRegex, RegexSplitPredicate.SplitType splitType) throws TextDBException {
ScanBasedSourceOperator scanSource = new ScanBasedSourceOperator(new ScanSourcePredicate(tableName));
RegexSplitOperator regexSplit = new RegexSplitOperator(new RegexSplitPredicate(splitRegex, splitAttrName, splitType));
regexSplit.setInputOperator(scanSource);
List<Tuple> results = new ArrayList<>();
regexSplit.open();
Tuple tuple;
while ((tuple = regexSplit.getNextTuple()) != null) {
results.add(tuple);
}
regexSplit.close();
return results;
}
use of edu.uci.ics.textdb.exp.source.scan.ScanSourcePredicate in project textdb by TextDB.
the class SamplerTest method containedInSamplerTable.
/*
* To test if all the sampled tuples are in the sampler table.
*/
public static boolean containedInSamplerTable(List<Tuple> sampleList) throws TextDBException {
ScanBasedSourceOperator scanSource = new ScanBasedSourceOperator(new ScanSourcePredicate(SAMPLER_TABLE));
scanSource.open();
Tuple nextTuple = null;
List<Tuple> returnedTuples = new ArrayList<Tuple>();
while ((nextTuple = scanSource.getNextTuple()) != null) {
returnedTuples.add(nextTuple);
}
scanSource.close();
boolean contains = TestUtils.containsAll(returnedTuples, sampleList);
return contains;
}
Aggregations