public void getNextTupleTestWithLimitOffset() throws Exception { List<Tuple> data = NlpEntityTestConstants.getOneSentenceTestTuple(); DataWriter oneSentenceDataWriter = RelationManager.getInstance().getTableDataWriter(ONE_SENTENCE_TABLE); oneSentenceDataWriter.open(); for (Tuple tuple : data) { oneSentenceDataWriter.insertTuple(tuple); } oneSentenceDataWriter.close(); String attribute1 = NlpEntityTestConstants.SENTENCE_ONE; List<String> attributeNames = Arrays.asList(attribute1); List<Tuple> returnedResults = getQueryResults(ONE_SENTENCE_TABLE, attributeNames, NlpEntityType.NE_ALL, 2, 2); List<Tuple> expectedResults = NlpEntityTestConstants.getTest10ResultTuples(); Assert.assertEquals(returnedResults.size(), 2); Assert.assertTrue(TestUtils.containsAll(expectedResults, returnedResults)); }
public void getNextTupleTestWithLimit() throws Exception { List<Tuple> data = NlpEntityTestConstants.getOneSentenceTestTuple(); DataWriter oneSentenceDataWriter = RelationManager.getInstance().getTableDataWriter(ONE_SENTENCE_TABLE); oneSentenceDataWriter.open(); for (Tuple tuple : data) { oneSentenceDataWriter.insertTuple(tuple); } oneSentenceDataWriter.close(); String attribute1 = NlpEntityTestConstants.SENTENCE_ONE; List<String> attributeNames = Arrays.asList(attribute1); List<Tuple> returnedResults = getQueryResults(ONE_SENTENCE_TABLE, attributeNames, NlpEntityType.NE_ALL, 3, 0); List<Tuple> expectedResults = NlpEntityTestConstants.getTest10ResultTuples(); // ExpectedResults is the array containing all the matches. // Since the order of returning records in returnedResults is not deterministic, we use containsAll // to ensure that the records in returnedResults are included in the ExpectedResults. Assert.assertEquals(returnedResults.size(), 3); Assert.assertTrue(TestUtils.containsAll(expectedResults, returnedResults)); }
@Test public void getNextTupleTest10() throws Exception { List<Tuple> data = NlpEntityTestConstants.getOneSentenceTestTuple(); DataWriter oneSentenceDataWriter = RelationManager.getInstance().getTableDataWriter(ONE_SENTENCE_TABLE); oneSentenceDataWriter.open(); for (Tuple tuple : data) { oneSentenceDataWriter.insertTuple(tuple); } oneSentenceDataWriter.close(); String attribute1 = NlpEntityTestConstants.SENTENCE_ONE; List<String> attributeNames = Arrays.asList(attribute1); List<Tuple> returnedResults = getQueryResults(ONE_SENTENCE_TABLE, attributeNames, NlpEntityType.NE_ALL); List<Tuple> expectedResults = NlpEntityTestConstants.getTest10ResultTuples(); boolean contains = TestUtils.equals(expectedResults, returnedResults); Assert.assertTrue(contains); }