@Test public void testMakeBag() throws ExecException { Tuple t1 = TupleFac.newTuple(new String[] {"Foo", "Bar"}); Tuple t2 = TupleFac.newTuple(new String[] {"Blue", "Green"}); DefaultDataBag bag = makeBag(new Tuple[] {t1, t2}); DefaultDataBag manualBag = new DefaultDataBag(); manualBag.add(t1); manualBag.add(t2); assertTrue(compareBags(manualBag, bag)); }
@Test public void testTrigrams() throws IOException { parmLen2.set(0, "This is a nice juicy test."); parmLen2.set(1, 3); // 3: trigrams DefaultDataBag resBag = makeBag( new Tuple[] { TupleFac.newTuple(new String[] {"nice,juicy,test"}), TupleFac.newTuple(new String[] {"This,is,a"}), TupleFac.newTuple(new String[] {"is,a,nice"}), TupleFac.newTuple(new String[] {"a,nice,juicy"}), }); assertTrue(compareBags(resBag, (DefaultDataBag) ngramFunc.exec(parmLen2))); }
@Test public void testBigrams() throws IOException { parmLen1.set(0, "This is a juicy test."); DefaultDataBag resBag = makeBag( new Tuple[] { TupleFac.newTuple(new String[] {"juicy,test"}), TupleFac.newTuple(new String[] {"is,a"}), TupleFac.newTuple(new String[] {"This,is"}), TupleFac.newTuple(new String[] {"a,juicy"}), }); assertTrue(compareBags(resBag, (DefaultDataBag) ngramFunc.exec(parmLen1))); }