Index: /trunk/autoquest-core-usageprofiles-test/src/test/java/de/ugoe/cs/autoquest/usageprofiles/TrieTest.java =================================================================== --- /trunk/autoquest-core-usageprofiles-test/src/test/java/de/ugoe/cs/autoquest/usageprofiles/TrieTest.java (revision 1251) +++ /trunk/autoquest-core-usageprofiles-test/src/test/java/de/ugoe/cs/autoquest/usageprofiles/TrieTest.java (revision 1252) @@ -31,6 +31,5 @@ /** - * The class TrieTest contains tests for the class - * {@link Trie}. + * The class TrieTest contains tests for the class {@link Trie}. * * @author Steffen Herbold @@ -39,1749 +38,1747 @@ public class TrieTest { - List sequence; - Collection symbols; - - private static void assertCollectionContent(Collection c1, - Collection c2) { - assertEquals(c1.size(), c2.size()); - for (Object obj : c1) { - assertTrue(c2.contains(obj)); - } - } - - @Test - public void testTrie_1() throws Exception { - - Trie result = new Trie(); - - assertNotNull(result); - assertEquals(0, result.getNumLeafs()); - assertEquals(0, result.getNumSymbols()); - assertEquals(0, result.getNumLeafAncestors()); - assertTrue(result.getKnownSymbols().isEmpty()); - } - - @Test - public void testTrie_2() throws Exception { - Trie trie1 = new Trie(); - trie1.train(sequence, 3); - - Trie result = new Trie(trie1); - - assertEquals(trie1, result); - assertNotSame(trie1, result); - } - - @Test(expected = java.lang.IllegalArgumentException.class) - public void testTrie_3() throws Exception { - new Trie((SymbolComparator) null); - } - - @Test(expected = java.lang.IllegalArgumentException.class) - public void testTrie_4() throws Exception { - new Trie((Trie) null); + List sequence; + Collection symbols; + + private static void assertCollectionContent(Collection c1, Collection c2) { + assertEquals(c1.size(), c2.size()); + for (Object obj : c1) { + assertTrue(c2.contains(obj)); } - - @Test - public void testAdd_1() throws Exception { - Trie fixture = new Trie(); - List seq = new ArrayList(); - seq.add("a"); - seq.add("b"); - - fixture.add(seq); - - assertEquals(1, fixture.getChild("a").getCount()); - assertEquals(1, fixture.getChild("a").getChild("b").getCount()); - assertNull(fixture.getChild("b")); - } - - @Test - public void testAdd_2() throws Exception { - Trie fixture = new Trie(); - - fixture.add(new ArrayList()); - - assertEquals(0, fixture.getNumSymbols()); - } - - @Test - public void testAdd_3() throws Exception { - Trie fixture = new Trie(); - - fixture.add(null); - - assertEquals(0, fixture.getNumSymbols()); - } - - @Test - public void testFind_1() throws Exception { - Trie fixture = new Trie(); - fixture.train(sequence, 3); - List findSequence = new ArrayList(); - findSequence.add("a"); - findSequence.add("b"); - findSequence.add("r"); - TrieNode expected = fixture.getChild("a").getChild("b") - .getChild("r"); - - TrieNode result = fixture.find(findSequence); - - assertEquals(expected, result); - } - - @Test - public void testFind_2() throws Exception { - Trie fixture = new Trie(); - fixture.train(sequence, 3); - List findSequence = new ArrayList(); - findSequence.add("c"); - findSequence.add("a"); - TrieNode expected = fixture.getChild("c").getChild("a"); - - TrieNode result = fixture.find(findSequence); - - assertEquals(expected, result); - } - - @Test - public void testFind_3() throws Exception { - Trie fixture = new Trie(); - fixture.train(sequence, 3); - List findSequence = new ArrayList(); - - TrieNode result = fixture.find(findSequence); - - assertTrue(result.isRoot()); - } - - @Test - public void testFind_4() throws Exception { - Trie fixture = new Trie(); - fixture.train(sequence, 3); - - TrieNode result = fixture.find(null); - - assertTrue(result.isRoot()); - } - - @Test - public void testGetChildCreate_1() throws Exception { - Trie fixture = new Trie(); - String symbol = "a"; - - TrieNode result = fixture.getChildCreate(symbol); - - assertEquals(symbol, result.getSymbol()); - assertEquals(0, result.getCount()); - assertTrue(result.isLeaf()); - } - - @Test(expected = java.lang.IllegalArgumentException.class) - public void testGetChildCreate_2() throws Exception { - Trie fixture = new Trie(); - fixture.getChildCreate(null); - } - - @Test - public void testGetContextSuffix_1() throws Exception { - Trie fixture = new Trie(); - fixture.train(sequence, 3); - List context = new ArrayList(); - context.add("a"); - context.add("a"); - context.add("b"); - List expected = new ArrayList(); - expected.add("a"); - expected.add("b"); - - List result = fixture.getContextSuffix(context); - - ListAssert.assertEquals(expected, result); - } - - @Test - public void testGetContextSuffix_2() throws Exception { - Trie fixture = new Trie(); - fixture.train(sequence, 3); - List context = new ArrayList(); - context.add("a"); - context.add("a"); - context.add("b"); - context.add("r"); - List expected = new ArrayList(); - expected.add("b"); - expected.add("r"); - - List result = fixture.getContextSuffix(context); - - ListAssert.assertEquals(expected, result); - } - - @Test - public void testGetContextSuffix_3() throws Exception { - Trie fixture = new Trie(); - fixture.train(sequence, 3); - List context = new ArrayList(); - context.add("a"); - context.add("a"); - context.add("b"); - context.add("x"); - List expected = new ArrayList(); - - List result = fixture.getContextSuffix(context); - - ListAssert.assertEquals(expected, result); - } - - @Test - public void testGetContextSuffix_4() throws Exception { - Trie fixture = new Trie(); - - List result = fixture.getContextSuffix(null); - - // add additional test code here - assertNotNull(result); - assertEquals(0, result.size()); - } - - @Test - public void testGetContextSuffix_5() throws Exception { - Trie fixture = new Trie(); - fixture.train(sequence, 3); - List context = new ArrayList(); - context.add("a"); - context.add("a"); - context.add("b"); - List expected = new ArrayList(); - expected.add("a"); - expected.add("b"); - - List result = fixture.getContextSuffix(context); - - ListAssert.assertEquals(expected, result); + } + + @Test + public void testTrie_1() throws Exception { + + Trie result = new Trie(); + + assertNotNull(result); + assertEquals(0, result.getNumLeafs()); + assertEquals(0, result.getNumSymbols()); + assertEquals(0, result.getNumLeafAncestors()); + assertTrue(result.getKnownSymbols().isEmpty()); + } + + @Test + public void testTrie_2() throws Exception { + Trie trie1 = new Trie(); + trie1.train(sequence, 3); + + Trie result = new Trie(trie1); + + assertEquals(trie1, result); + assertNotSame(trie1, result); + } + + @Test(expected = java.lang.IllegalArgumentException.class) + public void testTrie_3() throws Exception { + new Trie((SymbolComparator) null); + } + + @Test(expected = java.lang.IllegalArgumentException.class) + public void testTrie_4() throws Exception { + new Trie((Trie) null); + } + + @Test + public void testAdd_1() throws Exception { + Trie fixture = new Trie(); + List seq = new ArrayList(); + seq.add("a"); + seq.add("b"); + + fixture.add(seq); + + assertEquals(1, fixture.getChild("a").getCount()); + assertEquals(1, fixture.getChild("a").getChild("b").getCount()); + assertNull(fixture.getChild("b")); + } + + @Test + public void testAdd_2() throws Exception { + Trie fixture = new Trie(); + + fixture.add(new ArrayList()); + + assertEquals(0, fixture.getNumSymbols()); + } + + @Test + public void testAdd_3() throws Exception { + Trie fixture = new Trie(); + + fixture.add(null); + + assertEquals(0, fixture.getNumSymbols()); + } + + @Test + public void testFind_1() throws Exception { + Trie fixture = new Trie(); + fixture.train(sequence, 3); + List findSequence = new ArrayList(); + findSequence.add("a"); + findSequence.add("b"); + findSequence.add("r"); + TrieNode expected = fixture.getChild("a").getChild("b").getChild("r"); + + TrieNode result = fixture.find(findSequence); + + assertEquals(expected, result); + } + + @Test + public void testFind_2() throws Exception { + Trie fixture = new Trie(); + fixture.train(sequence, 3); + List findSequence = new ArrayList(); + findSequence.add("c"); + findSequence.add("a"); + TrieNode expected = fixture.getChild("c").getChild("a"); + + TrieNode result = fixture.find(findSequence); + + assertEquals(expected, result); + } + + @Test + public void testFind_3() throws Exception { + Trie fixture = new Trie(); + fixture.train(sequence, 3); + List findSequence = new ArrayList(); + + TrieNode result = fixture.find(findSequence); + + assertTrue(result.isRoot()); + } + + @Test + public void testFind_4() throws Exception { + Trie fixture = new Trie(); + fixture.train(sequence, 3); + + TrieNode result = fixture.find(null); + + assertTrue(result.isRoot()); + } + + @Test + public void testGetChildCreate_1() throws Exception { + Trie fixture = new Trie(); + String symbol = "a"; + + TrieNode result = fixture.getChildCreate(symbol); + + assertEquals(symbol, result.getSymbol()); + assertEquals(0, result.getCount()); + assertTrue(result.isLeaf()); + } + + @Test(expected = java.lang.IllegalArgumentException.class) + public void testGetChildCreate_2() throws Exception { + Trie fixture = new Trie(); + fixture.getChildCreate(null); + } + + @Test + public void testGetContextSuffix_1() throws Exception { + Trie fixture = new Trie(); + fixture.train(sequence, 3); + List context = new ArrayList(); + context.add("a"); + context.add("a"); + context.add("b"); + List expected = new ArrayList(); + expected.add("a"); + expected.add("b"); + + List result = fixture.getContextSuffix(context); + + ListAssert.assertEquals(expected, result); + } + + @Test + public void testGetContextSuffix_2() throws Exception { + Trie fixture = new Trie(); + fixture.train(sequence, 3); + List context = new ArrayList(); + context.add("a"); + context.add("a"); + context.add("b"); + context.add("r"); + List expected = new ArrayList(); + expected.add("b"); + expected.add("r"); + + List result = fixture.getContextSuffix(context); + + ListAssert.assertEquals(expected, result); + } + + @Test + public void testGetContextSuffix_3() throws Exception { + Trie fixture = new Trie(); + fixture.train(sequence, 3); + List context = new ArrayList(); + context.add("a"); + context.add("a"); + context.add("b"); + context.add("x"); + List expected = new ArrayList(); + + List result = fixture.getContextSuffix(context); + + ListAssert.assertEquals(expected, result); + } + + @Test + public void testGetContextSuffix_4() throws Exception { + Trie fixture = new Trie(); + + List result = fixture.getContextSuffix(null); + + // add additional test code here + assertNotNull(result); + assertEquals(0, result.size()); + } + + @Test + public void testGetContextSuffix_5() throws Exception { + Trie fixture = new Trie(); + fixture.train(sequence, 3); + List context = new ArrayList(); + context.add("a"); + context.add("a"); + context.add("b"); + List expected = new ArrayList(); + expected.add("a"); + expected.add("b"); + + List result = fixture.getContextSuffix(context); + + ListAssert.assertEquals(expected, result); + } + + @Test + public void testProcessWithTrieProcessor_1() throws Exception { + Trie fixture = new Trie(); + fixture.train(sequence, 2); + + final List sequences = new ArrayList(); + TrieProcessor processor = new TrieProcessor() { + @Override + public TrieProcessor.Result process(List sequence, int count) { + sequences.add(count + "_" + sequence.toString()); + return TrieProcessor.Result.CONTINUE; + } + + }; + + fixture.process(processor); + + List expected = new ArrayList(); + + expected.add("5_[a]"); + expected.add("2_[a, b]"); + expected.add("2_[b]"); + expected.add("2_[b, r]"); + expected.add("2_[r]"); + expected.add("2_[r, a]"); + expected.add("1_[a, c]"); + expected.add("1_[c]"); + expected.add("1_[c, a]"); + expected.add("1_[a, d]"); + expected.add("1_[d]"); + expected.add("1_[d, a]"); + + assertEquals(expected.size(), sequences.size()); + + for (String sequence : sequences) { + ListAssert.assertContains(expected, sequence); } - - @Test - public void testProcessWithTrieProcessor_1() throws Exception { - Trie fixture = new Trie(); - fixture.train(sequence, 2); - - final List sequences = new ArrayList(); - TrieProcessor processor = new TrieProcessor() { - @Override - public TrieProcessor.Result process(List sequence, int count) { - sequences.add(count + "_" + sequence.toString()); - return TrieProcessor.Result.CONTINUE; - } - - }; - - fixture.process(processor); - - List expected = new ArrayList(); - - expected.add("5_[a]"); - expected.add("2_[a, b]"); - expected.add("2_[b]"); - expected.add("2_[b, r]"); - expected.add("2_[r]"); - expected.add("2_[r, a]"); - expected.add("1_[a, c]"); - expected.add("1_[c]"); - expected.add("1_[c, a]"); - expected.add("1_[a, d]"); - expected.add("1_[d]"); - expected.add("1_[d, a]"); - - assertEquals(expected.size(), sequences.size()); - - for (String sequence : sequences) { - ListAssert.assertContains(expected, sequence); - } + } + + @Test + public void testProcessWithTrieProcessor_2() throws Exception { + Trie fixture = new Trie(); + fixture.train(sequence, 3); + + final List sequences = new ArrayList(); + TrieProcessor processor = new TrieProcessor() { + @Override + public TrieProcessor.Result process(List sequence, int count) { + sequences.add(count + "_" + sequence.toString()); + return TrieProcessor.Result.CONTINUE; + } + + }; + + fixture.process(processor); + + List expected = new ArrayList(); + + expected.add("5_[a]"); + expected.add("2_[a, b]"); + expected.add("2_[a, b, r]"); + expected.add("2_[b]"); + expected.add("2_[b, r]"); + expected.add("2_[b, r, a]"); + expected.add("2_[r]"); + expected.add("2_[r, a]"); + expected.add("1_[r, a, c]"); + expected.add("1_[a, c]"); + expected.add("1_[a, c, a]"); + expected.add("1_[c]"); + expected.add("1_[c, a]"); + expected.add("1_[c, a, d]"); + expected.add("1_[a, d]"); + expected.add("1_[a, d, a]"); + expected.add("1_[d]"); + expected.add("1_[d, a]"); + expected.add("1_[d, a, b]"); + + assertEquals(expected.size(), sequences.size()); + + for (String sequence : sequences) { + ListAssert.assertContains(expected, sequence); } - - @Test - public void testProcessWithTrieProcessor_2() throws Exception { - Trie fixture = new Trie(); - fixture.train(sequence, 3); - - final List sequences = new ArrayList(); - TrieProcessor processor = new TrieProcessor() { - @Override - public TrieProcessor.Result process(List sequence, int count) { - sequences.add(count + "_" + sequence.toString()); - return TrieProcessor.Result.CONTINUE; - } - - }; - - fixture.process(processor); - - List expected = new ArrayList(); - - expected.add("5_[a]"); - expected.add("2_[a, b]"); - expected.add("2_[a, b, r]"); - expected.add("2_[b]"); - expected.add("2_[b, r]"); - expected.add("2_[b, r, a]"); - expected.add("2_[r]"); - expected.add("2_[r, a]"); - expected.add("1_[r, a, c]"); - expected.add("1_[a, c]"); - expected.add("1_[a, c, a]"); - expected.add("1_[c]"); - expected.add("1_[c, a]"); - expected.add("1_[c, a, d]"); - expected.add("1_[a, d]"); - expected.add("1_[a, d, a]"); - expected.add("1_[d]"); - expected.add("1_[d, a]"); - expected.add("1_[d, a, b]"); - - assertEquals(expected.size(), sequences.size()); - - for (String sequence : sequences) { - ListAssert.assertContains(expected, sequence); - } + } + + @Test + public void testProcessWithTrieProcessor_3() throws Exception { + Trie fixture = new Trie(); + fixture.train(sequence, 4); + + final List sequences = new ArrayList(); + TrieProcessor processor = new TrieProcessor() { + @Override + public TrieProcessor.Result process(List sequence, int count) { + sequences.add(count + "_" + sequence.toString()); + return TrieProcessor.Result.CONTINUE; + } + + }; + + fixture.process(processor); + + List expected = new ArrayList(); + + expected.add("5_[a]"); + expected.add("2_[a, b]"); + expected.add("2_[a, b, r]"); + expected.add("2_[a, b, r, a]"); + expected.add("2_[b]"); + expected.add("2_[b, r]"); + expected.add("2_[b, r, a]"); + expected.add("1_[b, r, a, c]"); + expected.add("2_[r]"); + expected.add("2_[r, a]"); + expected.add("1_[r, a, c]"); + expected.add("1_[r, a, c, a]"); + expected.add("1_[a, c]"); + expected.add("1_[a, c, a]"); + expected.add("1_[a, c, a, d]"); + expected.add("1_[c]"); + expected.add("1_[c, a]"); + expected.add("1_[c, a, d]"); + expected.add("1_[c, a, d, a]"); + expected.add("1_[a, d]"); + expected.add("1_[a, d, a]"); + expected.add("1_[a, d, a, b]"); + expected.add("1_[d]"); + expected.add("1_[d, a]"); + expected.add("1_[d, a, b]"); + expected.add("1_[d, a, b, r]"); + + assertEquals(expected.size(), sequences.size()); + + for (String sequence : sequences) { + ListAssert.assertContains(expected, sequence); } - - @Test - public void testProcessWithTrieProcessor_3() throws Exception { - Trie fixture = new Trie(); - fixture.train(sequence, 4); - - final List sequences = new ArrayList(); - TrieProcessor processor = new TrieProcessor() { - @Override - public TrieProcessor.Result process(List sequence, int count) { - sequences.add(count + "_" + sequence.toString()); - return TrieProcessor.Result.CONTINUE; - } - - }; - - fixture.process(processor); - - List expected = new ArrayList(); - - expected.add("5_[a]"); - expected.add("2_[a, b]"); - expected.add("2_[a, b, r]"); - expected.add("2_[a, b, r, a]"); - expected.add("2_[b]"); - expected.add("2_[b, r]"); - expected.add("2_[b, r, a]"); - expected.add("1_[b, r, a, c]"); - expected.add("2_[r]"); - expected.add("2_[r, a]"); - expected.add("1_[r, a, c]"); - expected.add("1_[r, a, c, a]"); - expected.add("1_[a, c]"); - expected.add("1_[a, c, a]"); - expected.add("1_[a, c, a, d]"); - expected.add("1_[c]"); - expected.add("1_[c, a]"); - expected.add("1_[c, a, d]"); - expected.add("1_[c, a, d, a]"); - expected.add("1_[a, d]"); - expected.add("1_[a, d, a]"); - expected.add("1_[a, d, a, b]"); - expected.add("1_[d]"); - expected.add("1_[d, a]"); - expected.add("1_[d, a, b]"); - expected.add("1_[d, a, b, r]"); - - assertEquals(expected.size(), sequences.size()); - - for (String sequence : sequences) { - ListAssert.assertContains(expected, sequence); - } + } + + @Test + public void testProcessWithTrieProcessor_4() throws Exception { + Trie fixture = new Trie(); + fixture.train(sequence, 5); + + final List sequences = new ArrayList(); + TrieProcessor processor = new TrieProcessor() { + @Override + public TrieProcessor.Result process(List sequence, int count) { + sequences.add(count + "_" + sequence.toString()); + return TrieProcessor.Result.CONTINUE; + } + + }; + + fixture.process(processor); + + List expected = new ArrayList(); + + expected.add("5_[a]"); + expected.add("2_[a, b]"); + expected.add("2_[a, b, r]"); + expected.add("2_[a, b, r, a]"); + expected.add("1_[a, b, r, a, c]"); + expected.add("2_[b]"); + expected.add("2_[b, r]"); + expected.add("2_[b, r, a]"); + expected.add("1_[b, r, a, c]"); + expected.add("1_[b, r, a, c, a]"); + expected.add("2_[r]"); + expected.add("2_[r, a]"); + expected.add("1_[r, a, c]"); + expected.add("1_[r, a, c, a]"); + expected.add("1_[r, a, c, a, d]"); + expected.add("1_[a, c]"); + expected.add("1_[a, c, a]"); + expected.add("1_[a, c, a, d]"); + expected.add("1_[a, c, a, d, a]"); + expected.add("1_[c]"); + expected.add("1_[c, a]"); + expected.add("1_[c, a, d]"); + expected.add("1_[c, a, d, a]"); + expected.add("1_[c, a, d, a, b]"); + expected.add("1_[a, d]"); + expected.add("1_[a, d, a]"); + expected.add("1_[a, d, a, b]"); + expected.add("1_[a, d, a, b, r]"); + expected.add("1_[d]"); + expected.add("1_[d, a]"); + expected.add("1_[d, a, b]"); + expected.add("1_[d, a, b, r]"); + expected.add("1_[d, a, b, r, a]"); + + assertEquals(expected.size(), sequences.size()); + + for (String sequence : sequences) { + ListAssert.assertContains(expected, sequence); } - - @Test - public void testProcessWithTrieProcessor_4() throws Exception { - Trie fixture = new Trie(); - fixture.train(sequence, 5); - - final List sequences = new ArrayList(); - TrieProcessor processor = new TrieProcessor() { - @Override - public TrieProcessor.Result process(List sequence, int count) { - sequences.add(count + "_" + sequence.toString()); - return TrieProcessor.Result.CONTINUE; - } - - }; - - fixture.process(processor); - - List expected = new ArrayList(); - - expected.add("5_[a]"); - expected.add("2_[a, b]"); - expected.add("2_[a, b, r]"); - expected.add("2_[a, b, r, a]"); - expected.add("1_[a, b, r, a, c]"); - expected.add("2_[b]"); - expected.add("2_[b, r]"); - expected.add("2_[b, r, a]"); - expected.add("1_[b, r, a, c]"); - expected.add("1_[b, r, a, c, a]"); - expected.add("2_[r]"); - expected.add("2_[r, a]"); - expected.add("1_[r, a, c]"); - expected.add("1_[r, a, c, a]"); - expected.add("1_[r, a, c, a, d]"); - expected.add("1_[a, c]"); - expected.add("1_[a, c, a]"); - expected.add("1_[a, c, a, d]"); - expected.add("1_[a, c, a, d, a]"); - expected.add("1_[c]"); - expected.add("1_[c, a]"); - expected.add("1_[c, a, d]"); - expected.add("1_[c, a, d, a]"); - expected.add("1_[c, a, d, a, b]"); - expected.add("1_[a, d]"); - expected.add("1_[a, d, a]"); - expected.add("1_[a, d, a, b]"); - expected.add("1_[a, d, a, b, r]"); - expected.add("1_[d]"); - expected.add("1_[d, a]"); - expected.add("1_[d, a, b]"); - expected.add("1_[d, a, b, r]"); - expected.add("1_[d, a, b, r, a]"); - - assertEquals(expected.size(), sequences.size()); - - for (String sequence : sequences) { - ListAssert.assertContains(expected, sequence); - } + } + + @Test + public void testProcessWithTrieProcessor_5() throws Exception { + Trie fixture = new Trie(); + fixture.train(sequence, 6); + + final List sequences = new ArrayList(); + TrieProcessor processor = new TrieProcessor() { + @Override + public TrieProcessor.Result process(List sequence, int count) { + sequences.add(count + "_" + sequence.toString()); + return TrieProcessor.Result.CONTINUE; + } + + }; + + fixture.process(processor); + + List expected = new ArrayList(); + + expected.add("5_[a]"); + expected.add("2_[a, b]"); + expected.add("2_[a, b, r]"); + expected.add("2_[a, b, r, a]"); + expected.add("1_[a, b, r, a, c]"); + expected.add("1_[a, b, r, a, c, a]"); + expected.add("2_[b]"); + expected.add("2_[b, r]"); + expected.add("2_[b, r, a]"); + expected.add("1_[b, r, a, c]"); + expected.add("1_[b, r, a, c, a]"); + expected.add("1_[b, r, a, c, a, d]"); + expected.add("2_[r]"); + expected.add("2_[r, a]"); + expected.add("1_[r, a, c]"); + expected.add("1_[r, a, c, a]"); + expected.add("1_[r, a, c, a, d]"); + expected.add("1_[r, a, c, a, d, a]"); + expected.add("1_[a, c]"); + expected.add("1_[a, c, a]"); + expected.add("1_[a, c, a, d]"); + expected.add("1_[a, c, a, d, a]"); + expected.add("1_[a, c, a, d, a, b]"); + expected.add("1_[c]"); + expected.add("1_[c, a]"); + expected.add("1_[c, a, d]"); + expected.add("1_[c, a, d, a]"); + expected.add("1_[c, a, d, a, b]"); + expected.add("1_[c, a, d, a, b, r]"); + expected.add("1_[a, d]"); + expected.add("1_[a, d, a]"); + expected.add("1_[a, d, a, b]"); + expected.add("1_[a, d, a, b, r]"); + expected.add("1_[a, d, a, b, r, a]"); + expected.add("1_[d]"); + expected.add("1_[d, a]"); + expected.add("1_[d, a, b]"); + expected.add("1_[d, a, b, r]"); + expected.add("1_[d, a, b, r, a]"); + + assertEquals(expected.size(), sequences.size()); + + for (String sequence : sequences) { + ListAssert.assertContains(expected, sequence); } - - @Test - public void testProcessWithTrieProcessor_5() throws Exception { - Trie fixture = new Trie(); - fixture.train(sequence, 6); - - final List sequences = new ArrayList(); - TrieProcessor processor = new TrieProcessor() { - @Override - public TrieProcessor.Result process(List sequence, int count) { - sequences.add(count + "_" + sequence.toString()); - return TrieProcessor.Result.CONTINUE; - } - - }; - - fixture.process(processor); - - List expected = new ArrayList(); - - expected.add("5_[a]"); - expected.add("2_[a, b]"); - expected.add("2_[a, b, r]"); - expected.add("2_[a, b, r, a]"); - expected.add("1_[a, b, r, a, c]"); - expected.add("1_[a, b, r, a, c, a]"); - expected.add("2_[b]"); - expected.add("2_[b, r]"); - expected.add("2_[b, r, a]"); - expected.add("1_[b, r, a, c]"); - expected.add("1_[b, r, a, c, a]"); - expected.add("1_[b, r, a, c, a, d]"); - expected.add("2_[r]"); - expected.add("2_[r, a]"); - expected.add("1_[r, a, c]"); - expected.add("1_[r, a, c, a]"); - expected.add("1_[r, a, c, a, d]"); - expected.add("1_[r, a, c, a, d, a]"); - expected.add("1_[a, c]"); - expected.add("1_[a, c, a]"); - expected.add("1_[a, c, a, d]"); - expected.add("1_[a, c, a, d, a]"); - expected.add("1_[a, c, a, d, a, b]"); - expected.add("1_[c]"); - expected.add("1_[c, a]"); - expected.add("1_[c, a, d]"); - expected.add("1_[c, a, d, a]"); - expected.add("1_[c, a, d, a, b]"); - expected.add("1_[c, a, d, a, b, r]"); - expected.add("1_[a, d]"); - expected.add("1_[a, d, a]"); - expected.add("1_[a, d, a, b]"); - expected.add("1_[a, d, a, b, r]"); - expected.add("1_[a, d, a, b, r, a]"); - expected.add("1_[d]"); - expected.add("1_[d, a]"); - expected.add("1_[d, a, b]"); - expected.add("1_[d, a, b, r]"); - expected.add("1_[d, a, b, r, a]"); - - assertEquals(expected.size(), sequences.size()); - - for (String sequence : sequences) { - ListAssert.assertContains(expected, sequence); - } + } + + @Test + public void testProcessWithTrieProcessor_6() throws Exception { + Trie fixture = new Trie(); + fixture.train(sequence, 7); + + final List sequences = new ArrayList(); + TrieProcessor processor = new TrieProcessor() { + @Override + public TrieProcessor.Result process(List sequence, int count) { + sequences.add(count + "_" + sequence.toString()); + return TrieProcessor.Result.CONTINUE; + } + + }; + + fixture.process(processor); + + List expected = new ArrayList(); + + expected.add("5_[a]"); + expected.add("2_[a, b]"); + expected.add("2_[a, b, r]"); + expected.add("2_[a, b, r, a]"); + expected.add("1_[a, b, r, a, c]"); + expected.add("1_[a, b, r, a, c, a]"); + expected.add("1_[a, b, r, a, c, a, d]"); + expected.add("2_[b]"); + expected.add("2_[b, r]"); + expected.add("2_[b, r, a]"); + expected.add("1_[b, r, a, c]"); + expected.add("1_[b, r, a, c, a]"); + expected.add("1_[b, r, a, c, a, d]"); + expected.add("1_[b, r, a, c, a, d, a]"); + expected.add("2_[r]"); + expected.add("2_[r, a]"); + expected.add("1_[r, a, c]"); + expected.add("1_[r, a, c, a]"); + expected.add("1_[r, a, c, a, d]"); + expected.add("1_[r, a, c, a, d, a]"); + expected.add("1_[r, a, c, a, d, a, b]"); + expected.add("1_[a, c]"); + expected.add("1_[a, c, a]"); + expected.add("1_[a, c, a, d]"); + expected.add("1_[a, c, a, d, a]"); + expected.add("1_[a, c, a, d, a, b]"); + expected.add("1_[a, c, a, d, a, b, r]"); + expected.add("1_[c]"); + expected.add("1_[c, a]"); + expected.add("1_[c, a, d]"); + expected.add("1_[c, a, d, a]"); + expected.add("1_[c, a, d, a, b]"); + expected.add("1_[c, a, d, a, b, r]"); + expected.add("1_[c, a, d, a, b, r, a]"); + expected.add("1_[a, d]"); + expected.add("1_[a, d, a]"); + expected.add("1_[a, d, a, b]"); + expected.add("1_[a, d, a, b, r]"); + expected.add("1_[a, d, a, b, r, a]"); + expected.add("1_[d]"); + expected.add("1_[d, a]"); + expected.add("1_[d, a, b]"); + expected.add("1_[d, a, b, r]"); + expected.add("1_[d, a, b, r, a]"); + + assertEquals(expected.size(), sequences.size()); + + for (String sequence : sequences) { + ListAssert.assertContains(expected, sequence); } - - @Test - public void testProcessWithTrieProcessor_6() throws Exception { - Trie fixture = new Trie(); - fixture.train(sequence, 7); - - final List sequences = new ArrayList(); - TrieProcessor processor = new TrieProcessor() { - @Override - public TrieProcessor.Result process(List sequence, int count) { - sequences.add(count + "_" + sequence.toString()); - return TrieProcessor.Result.CONTINUE; - } - - }; - - fixture.process(processor); - - List expected = new ArrayList(); - - expected.add("5_[a]"); - expected.add("2_[a, b]"); - expected.add("2_[a, b, r]"); - expected.add("2_[a, b, r, a]"); - expected.add("1_[a, b, r, a, c]"); - expected.add("1_[a, b, r, a, c, a]"); - expected.add("1_[a, b, r, a, c, a, d]"); - expected.add("2_[b]"); - expected.add("2_[b, r]"); - expected.add("2_[b, r, a]"); - expected.add("1_[b, r, a, c]"); - expected.add("1_[b, r, a, c, a]"); - expected.add("1_[b, r, a, c, a, d]"); - expected.add("1_[b, r, a, c, a, d, a]"); - expected.add("2_[r]"); - expected.add("2_[r, a]"); - expected.add("1_[r, a, c]"); - expected.add("1_[r, a, c, a]"); - expected.add("1_[r, a, c, a, d]"); - expected.add("1_[r, a, c, a, d, a]"); - expected.add("1_[r, a, c, a, d, a, b]"); - expected.add("1_[a, c]"); - expected.add("1_[a, c, a]"); - expected.add("1_[a, c, a, d]"); - expected.add("1_[a, c, a, d, a]"); - expected.add("1_[a, c, a, d, a, b]"); - expected.add("1_[a, c, a, d, a, b, r]"); - expected.add("1_[c]"); - expected.add("1_[c, a]"); - expected.add("1_[c, a, d]"); - expected.add("1_[c, a, d, a]"); - expected.add("1_[c, a, d, a, b]"); - expected.add("1_[c, a, d, a, b, r]"); - expected.add("1_[c, a, d, a, b, r, a]"); - expected.add("1_[a, d]"); - expected.add("1_[a, d, a]"); - expected.add("1_[a, d, a, b]"); - expected.add("1_[a, d, a, b, r]"); - expected.add("1_[a, d, a, b, r, a]"); - expected.add("1_[d]"); - expected.add("1_[d, a]"); - expected.add("1_[d, a, b]"); - expected.add("1_[d, a, b, r]"); - expected.add("1_[d, a, b, r, a]"); - - assertEquals(expected.size(), sequences.size()); - - for (String sequence : sequences) { - ListAssert.assertContains(expected, sequence); - } + } + + @Test + public void testProcessWithTrieProcessor_7() throws Exception { + Trie fixture = new Trie(); + fixture.train(sequence, 8); + + final List sequences = new ArrayList(); + TrieProcessor processor = new TrieProcessor() { + @Override + public TrieProcessor.Result process(List sequence, int count) { + sequences.add(count + "_" + sequence.toString()); + return TrieProcessor.Result.CONTINUE; + } + + }; + + fixture.process(processor); + + List expected = new ArrayList(); + + expected.add("5_[a]"); + expected.add("2_[a, b]"); + expected.add("2_[a, b, r]"); + expected.add("2_[a, b, r, a]"); + expected.add("1_[a, b, r, a, c]"); + expected.add("1_[a, b, r, a, c, a]"); + expected.add("1_[a, b, r, a, c, a, d]"); + expected.add("1_[a, b, r, a, c, a, d, a]"); + expected.add("2_[b]"); + expected.add("2_[b, r]"); + expected.add("2_[b, r, a]"); + expected.add("1_[b, r, a, c]"); + expected.add("1_[b, r, a, c, a]"); + expected.add("1_[b, r, a, c, a, d]"); + expected.add("1_[b, r, a, c, a, d, a]"); + expected.add("1_[b, r, a, c, a, d, a, b]"); + expected.add("2_[r]"); + expected.add("2_[r, a]"); + expected.add("1_[r, a, c]"); + expected.add("1_[r, a, c, a]"); + expected.add("1_[r, a, c, a, d]"); + expected.add("1_[r, a, c, a, d, a]"); + expected.add("1_[r, a, c, a, d, a, b]"); + expected.add("1_[r, a, c, a, d, a, b, r]"); + expected.add("1_[a, c]"); + expected.add("1_[a, c, a]"); + expected.add("1_[a, c, a, d]"); + expected.add("1_[a, c, a, d, a]"); + expected.add("1_[a, c, a, d, a, b]"); + expected.add("1_[a, c, a, d, a, b, r]"); + expected.add("1_[a, c, a, d, a, b, r, a]"); + expected.add("1_[c]"); + expected.add("1_[c, a]"); + expected.add("1_[c, a, d]"); + expected.add("1_[c, a, d, a]"); + expected.add("1_[c, a, d, a, b]"); + expected.add("1_[c, a, d, a, b, r]"); + expected.add("1_[c, a, d, a, b, r, a]"); + expected.add("1_[a, d]"); + expected.add("1_[a, d, a]"); + expected.add("1_[a, d, a, b]"); + expected.add("1_[a, d, a, b, r]"); + expected.add("1_[a, d, a, b, r, a]"); + expected.add("1_[d]"); + expected.add("1_[d, a]"); + expected.add("1_[d, a, b]"); + expected.add("1_[d, a, b, r]"); + expected.add("1_[d, a, b, r, a]"); + + assertEquals(expected.size(), sequences.size()); + + for (String sequence : sequences) { + ListAssert.assertContains(expected, sequence); } - - @Test - public void testProcessWithTrieProcessor_7() throws Exception { - Trie fixture = new Trie(); - fixture.train(sequence, 8); - - final List sequences = new ArrayList(); - TrieProcessor processor = new TrieProcessor() { - @Override - public TrieProcessor.Result process(List sequence, int count) { - sequences.add(count + "_" + sequence.toString()); - return TrieProcessor.Result.CONTINUE; - } - - }; - - fixture.process(processor); - - List expected = new ArrayList(); - - expected.add("5_[a]"); - expected.add("2_[a, b]"); - expected.add("2_[a, b, r]"); - expected.add("2_[a, b, r, a]"); - expected.add("1_[a, b, r, a, c]"); - expected.add("1_[a, b, r, a, c, a]"); - expected.add("1_[a, b, r, a, c, a, d]"); - expected.add("1_[a, b, r, a, c, a, d, a]"); - expected.add("2_[b]"); - expected.add("2_[b, r]"); - expected.add("2_[b, r, a]"); - expected.add("1_[b, r, a, c]"); - expected.add("1_[b, r, a, c, a]"); - expected.add("1_[b, r, a, c, a, d]"); - expected.add("1_[b, r, a, c, a, d, a]"); - expected.add("1_[b, r, a, c, a, d, a, b]"); - expected.add("2_[r]"); - expected.add("2_[r, a]"); - expected.add("1_[r, a, c]"); - expected.add("1_[r, a, c, a]"); - expected.add("1_[r, a, c, a, d]"); - expected.add("1_[r, a, c, a, d, a]"); - expected.add("1_[r, a, c, a, d, a, b]"); - expected.add("1_[r, a, c, a, d, a, b, r]"); - expected.add("1_[a, c]"); - expected.add("1_[a, c, a]"); - expected.add("1_[a, c, a, d]"); - expected.add("1_[a, c, a, d, a]"); - expected.add("1_[a, c, a, d, a, b]"); - expected.add("1_[a, c, a, d, a, b, r]"); - expected.add("1_[a, c, a, d, a, b, r, a]"); - expected.add("1_[c]"); - expected.add("1_[c, a]"); - expected.add("1_[c, a, d]"); - expected.add("1_[c, a, d, a]"); - expected.add("1_[c, a, d, a, b]"); - expected.add("1_[c, a, d, a, b, r]"); - expected.add("1_[c, a, d, a, b, r, a]"); - expected.add("1_[a, d]"); - expected.add("1_[a, d, a]"); - expected.add("1_[a, d, a, b]"); - expected.add("1_[a, d, a, b, r]"); - expected.add("1_[a, d, a, b, r, a]"); - expected.add("1_[d]"); - expected.add("1_[d, a]"); - expected.add("1_[d, a, b]"); - expected.add("1_[d, a, b, r]"); - expected.add("1_[d, a, b, r, a]"); - - assertEquals(expected.size(), sequences.size()); - - for (String sequence : sequences) { - ListAssert.assertContains(expected, sequence); - } + } + + @Test + public void testProcessWithTrieProcessor_8() throws Exception { + Trie fixture = new Trie(); + fixture.train(sequence, 9); + + final List sequences = new ArrayList(); + TrieProcessor processor = new TrieProcessor() { + @Override + public TrieProcessor.Result process(List sequence, int count) { + sequences.add(count + "_" + sequence.toString()); + return TrieProcessor.Result.CONTINUE; + } + + }; + + fixture.process(processor); + + List expected = new ArrayList(); + + expected.add("5_[a]"); + expected.add("2_[a, b]"); + expected.add("2_[a, b, r]"); + expected.add("2_[a, b, r, a]"); + expected.add("1_[a, b, r, a, c]"); + expected.add("1_[a, b, r, a, c, a]"); + expected.add("1_[a, b, r, a, c, a, d]"); + expected.add("1_[a, b, r, a, c, a, d, a]"); + expected.add("1_[a, b, r, a, c, a, d, a, b]"); + expected.add("2_[b]"); + expected.add("2_[b, r]"); + expected.add("2_[b, r, a]"); + expected.add("1_[b, r, a, c]"); + expected.add("1_[b, r, a, c, a]"); + expected.add("1_[b, r, a, c, a, d]"); + expected.add("1_[b, r, a, c, a, d, a]"); + expected.add("1_[b, r, a, c, a, d, a, b]"); + expected.add("1_[b, r, a, c, a, d, a, b, r]"); + expected.add("2_[r]"); + expected.add("2_[r, a]"); + expected.add("1_[r, a, c]"); + expected.add("1_[r, a, c, a]"); + expected.add("1_[r, a, c, a, d]"); + expected.add("1_[r, a, c, a, d, a]"); + expected.add("1_[r, a, c, a, d, a, b]"); + expected.add("1_[r, a, c, a, d, a, b, r]"); + expected.add("1_[r, a, c, a, d, a, b, r, a]"); + expected.add("1_[a, c]"); + expected.add("1_[a, c, a]"); + expected.add("1_[a, c, a, d]"); + expected.add("1_[a, c, a, d, a]"); + expected.add("1_[a, c, a, d, a, b]"); + expected.add("1_[a, c, a, d, a, b, r]"); + expected.add("1_[a, c, a, d, a, b, r, a]"); + expected.add("1_[c]"); + expected.add("1_[c, a]"); + expected.add("1_[c, a, d]"); + expected.add("1_[c, a, d, a]"); + expected.add("1_[c, a, d, a, b]"); + expected.add("1_[c, a, d, a, b, r]"); + expected.add("1_[c, a, d, a, b, r, a]"); + expected.add("1_[a, d]"); + expected.add("1_[a, d, a]"); + expected.add("1_[a, d, a, b]"); + expected.add("1_[a, d, a, b, r]"); + expected.add("1_[a, d, a, b, r, a]"); + expected.add("1_[d]"); + expected.add("1_[d, a]"); + expected.add("1_[d, a, b]"); + expected.add("1_[d, a, b, r]"); + expected.add("1_[d, a, b, r, a]"); + + assertEquals(expected.size(), sequences.size()); + + for (String sequence : sequences) { + ListAssert.assertContains(expected, sequence); } - - @Test - public void testProcessWithTrieProcessor_8() throws Exception { - Trie fixture = new Trie(); - fixture.train(sequence, 9); - - final List sequences = new ArrayList(); - TrieProcessor processor = new TrieProcessor() { - @Override - public TrieProcessor.Result process(List sequence, int count) { - sequences.add(count + "_" + sequence.toString()); - return TrieProcessor.Result.CONTINUE; - } - - }; - - fixture.process(processor); - - List expected = new ArrayList(); - - expected.add("5_[a]"); - expected.add("2_[a, b]"); - expected.add("2_[a, b, r]"); - expected.add("2_[a, b, r, a]"); - expected.add("1_[a, b, r, a, c]"); - expected.add("1_[a, b, r, a, c, a]"); - expected.add("1_[a, b, r, a, c, a, d]"); - expected.add("1_[a, b, r, a, c, a, d, a]"); - expected.add("1_[a, b, r, a, c, a, d, a, b]"); - expected.add("2_[b]"); - expected.add("2_[b, r]"); - expected.add("2_[b, r, a]"); - expected.add("1_[b, r, a, c]"); - expected.add("1_[b, r, a, c, a]"); - expected.add("1_[b, r, a, c, a, d]"); - expected.add("1_[b, r, a, c, a, d, a]"); - expected.add("1_[b, r, a, c, a, d, a, b]"); - expected.add("1_[b, r, a, c, a, d, a, b, r]"); - expected.add("2_[r]"); - expected.add("2_[r, a]"); - expected.add("1_[r, a, c]"); - expected.add("1_[r, a, c, a]"); - expected.add("1_[r, a, c, a, d]"); - expected.add("1_[r, a, c, a, d, a]"); - expected.add("1_[r, a, c, a, d, a, b]"); - expected.add("1_[r, a, c, a, d, a, b, r]"); - expected.add("1_[r, a, c, a, d, a, b, r, a]"); - expected.add("1_[a, c]"); - expected.add("1_[a, c, a]"); - expected.add("1_[a, c, a, d]"); - expected.add("1_[a, c, a, d, a]"); - expected.add("1_[a, c, a, d, a, b]"); - expected.add("1_[a, c, a, d, a, b, r]"); - expected.add("1_[a, c, a, d, a, b, r, a]"); - expected.add("1_[c]"); - expected.add("1_[c, a]"); - expected.add("1_[c, a, d]"); - expected.add("1_[c, a, d, a]"); - expected.add("1_[c, a, d, a, b]"); - expected.add("1_[c, a, d, a, b, r]"); - expected.add("1_[c, a, d, a, b, r, a]"); - expected.add("1_[a, d]"); - expected.add("1_[a, d, a]"); - expected.add("1_[a, d, a, b]"); - expected.add("1_[a, d, a, b, r]"); - expected.add("1_[a, d, a, b, r, a]"); - expected.add("1_[d]"); - expected.add("1_[d, a]"); - expected.add("1_[d, a, b]"); - expected.add("1_[d, a, b, r]"); - expected.add("1_[d, a, b, r, a]"); - - assertEquals(expected.size(), sequences.size()); - - for (String sequence : sequences) { - ListAssert.assertContains(expected, sequence); - } + } + + @Test + public void testProcessWithTrieProcessor_9() throws Exception { + Trie fixture = new Trie(); + fixture.train(sequence, 10); + + final List sequences = new ArrayList(); + TrieProcessor processor = new TrieProcessor() { + @Override + public TrieProcessor.Result process(List sequence, int count) { + sequences.add(count + "_" + sequence.toString()); + return TrieProcessor.Result.CONTINUE; + } + + }; + + fixture.process(processor); + + List expected = new ArrayList(); + + expected.add("5_[a]"); + expected.add("2_[a, b]"); + expected.add("2_[a, b, r]"); + expected.add("2_[a, b, r, a]"); + expected.add("1_[a, b, r, a, c]"); + expected.add("1_[a, b, r, a, c, a]"); + expected.add("1_[a, b, r, a, c, a, d]"); + expected.add("1_[a, b, r, a, c, a, d, a]"); + expected.add("1_[a, b, r, a, c, a, d, a, b]"); + expected.add("1_[a, b, r, a, c, a, d, a, b, r]"); + expected.add("2_[b]"); + expected.add("2_[b, r]"); + expected.add("2_[b, r, a]"); + expected.add("1_[b, r, a, c]"); + expected.add("1_[b, r, a, c, a]"); + expected.add("1_[b, r, a, c, a, d]"); + expected.add("1_[b, r, a, c, a, d, a]"); + expected.add("1_[b, r, a, c, a, d, a, b]"); + expected.add("1_[b, r, a, c, a, d, a, b, r]"); + expected.add("1_[b, r, a, c, a, d, a, b, r, a]"); + expected.add("2_[r]"); + expected.add("2_[r, a]"); + expected.add("1_[r, a, c]"); + expected.add("1_[r, a, c, a]"); + expected.add("1_[r, a, c, a, d]"); + expected.add("1_[r, a, c, a, d, a]"); + expected.add("1_[r, a, c, a, d, a, b]"); + expected.add("1_[r, a, c, a, d, a, b, r]"); + expected.add("1_[r, a, c, a, d, a, b, r, a]"); + expected.add("1_[a, c]"); + expected.add("1_[a, c, a]"); + expected.add("1_[a, c, a, d]"); + expected.add("1_[a, c, a, d, a]"); + expected.add("1_[a, c, a, d, a, b]"); + expected.add("1_[a, c, a, d, a, b, r]"); + expected.add("1_[a, c, a, d, a, b, r, a]"); + expected.add("1_[c]"); + expected.add("1_[c, a]"); + expected.add("1_[c, a, d]"); + expected.add("1_[c, a, d, a]"); + expected.add("1_[c, a, d, a, b]"); + expected.add("1_[c, a, d, a, b, r]"); + expected.add("1_[c, a, d, a, b, r, a]"); + expected.add("1_[a, d]"); + expected.add("1_[a, d, a]"); + expected.add("1_[a, d, a, b]"); + expected.add("1_[a, d, a, b, r]"); + expected.add("1_[a, d, a, b, r, a]"); + expected.add("1_[d]"); + expected.add("1_[d, a]"); + expected.add("1_[d, a, b]"); + expected.add("1_[d, a, b, r]"); + expected.add("1_[d, a, b, r, a]"); + + assertEquals(expected.size(), sequences.size()); + + for (String sequence : sequences) { + ListAssert.assertContains(expected, sequence); } - - @Test - public void testProcessWithTrieProcessor_9() throws Exception { - Trie fixture = new Trie(); - fixture.train(sequence, 10); - - final List sequences = new ArrayList(); - TrieProcessor processor = new TrieProcessor() { - @Override - public TrieProcessor.Result process(List sequence, int count) { - sequences.add(count + "_" + sequence.toString()); - return TrieProcessor.Result.CONTINUE; - } - - }; - - fixture.process(processor); - - List expected = new ArrayList(); - - expected.add("5_[a]"); - expected.add("2_[a, b]"); - expected.add("2_[a, b, r]"); - expected.add("2_[a, b, r, a]"); - expected.add("1_[a, b, r, a, c]"); - expected.add("1_[a, b, r, a, c, a]"); - expected.add("1_[a, b, r, a, c, a, d]"); - expected.add("1_[a, b, r, a, c, a, d, a]"); - expected.add("1_[a, b, r, a, c, a, d, a, b]"); - expected.add("1_[a, b, r, a, c, a, d, a, b, r]"); - expected.add("2_[b]"); - expected.add("2_[b, r]"); - expected.add("2_[b, r, a]"); - expected.add("1_[b, r, a, c]"); - expected.add("1_[b, r, a, c, a]"); - expected.add("1_[b, r, a, c, a, d]"); - expected.add("1_[b, r, a, c, a, d, a]"); - expected.add("1_[b, r, a, c, a, d, a, b]"); - expected.add("1_[b, r, a, c, a, d, a, b, r]"); - expected.add("1_[b, r, a, c, a, d, a, b, r, a]"); - expected.add("2_[r]"); - expected.add("2_[r, a]"); - expected.add("1_[r, a, c]"); - expected.add("1_[r, a, c, a]"); - expected.add("1_[r, a, c, a, d]"); - expected.add("1_[r, a, c, a, d, a]"); - expected.add("1_[r, a, c, a, d, a, b]"); - expected.add("1_[r, a, c, a, d, a, b, r]"); - expected.add("1_[r, a, c, a, d, a, b, r, a]"); - expected.add("1_[a, c]"); - expected.add("1_[a, c, a]"); - expected.add("1_[a, c, a, d]"); - expected.add("1_[a, c, a, d, a]"); - expected.add("1_[a, c, a, d, a, b]"); - expected.add("1_[a, c, a, d, a, b, r]"); - expected.add("1_[a, c, a, d, a, b, r, a]"); - expected.add("1_[c]"); - expected.add("1_[c, a]"); - expected.add("1_[c, a, d]"); - expected.add("1_[c, a, d, a]"); - expected.add("1_[c, a, d, a, b]"); - expected.add("1_[c, a, d, a, b, r]"); - expected.add("1_[c, a, d, a, b, r, a]"); - expected.add("1_[a, d]"); - expected.add("1_[a, d, a]"); - expected.add("1_[a, d, a, b]"); - expected.add("1_[a, d, a, b, r]"); - expected.add("1_[a, d, a, b, r, a]"); - expected.add("1_[d]"); - expected.add("1_[d, a]"); - expected.add("1_[d, a, b]"); - expected.add("1_[d, a, b, r]"); - expected.add("1_[d, a, b, r, a]"); - - assertEquals(expected.size(), sequences.size()); - - for (String sequence : sequences) { - ListAssert.assertContains(expected, sequence); - } + } + + @Test + public void testProcessWithTrieProcessor_10() throws Exception { + Trie fixture = new Trie(); + fixture.train(sequence, 11); + + final List sequences = new ArrayList(); + TrieProcessor processor = new TrieProcessor() { + @Override + public TrieProcessor.Result process(List sequence, int count) { + sequences.add(count + "_" + sequence.toString()); + return TrieProcessor.Result.CONTINUE; + } + + }; + + fixture.process(processor); + + List expected = new ArrayList(); + + expected.add("5_[a]"); + expected.add("2_[a, b]"); + expected.add("2_[a, b, r]"); + expected.add("2_[a, b, r, a]"); + expected.add("1_[a, b, r, a, c]"); + expected.add("1_[a, b, r, a, c, a]"); + expected.add("1_[a, b, r, a, c, a, d]"); + expected.add("1_[a, b, r, a, c, a, d, a]"); + expected.add("1_[a, b, r, a, c, a, d, a, b]"); + expected.add("1_[a, b, r, a, c, a, d, a, b, r]"); + expected.add("1_[a, b, r, a, c, a, d, a, b, r, a]"); + expected.add("2_[b]"); + expected.add("2_[b, r]"); + expected.add("2_[b, r, a]"); + expected.add("1_[b, r, a, c]"); + expected.add("1_[b, r, a, c, a]"); + expected.add("1_[b, r, a, c, a, d]"); + expected.add("1_[b, r, a, c, a, d, a]"); + expected.add("1_[b, r, a, c, a, d, a, b]"); + expected.add("1_[b, r, a, c, a, d, a, b, r]"); + expected.add("1_[b, r, a, c, a, d, a, b, r, a]"); + expected.add("2_[r]"); + expected.add("2_[r, a]"); + expected.add("1_[r, a, c]"); + expected.add("1_[r, a, c, a]"); + expected.add("1_[r, a, c, a, d]"); + expected.add("1_[r, a, c, a, d, a]"); + expected.add("1_[r, a, c, a, d, a, b]"); + expected.add("1_[r, a, c, a, d, a, b, r]"); + expected.add("1_[r, a, c, a, d, a, b, r, a]"); + expected.add("1_[a, c]"); + expected.add("1_[a, c, a]"); + expected.add("1_[a, c, a, d]"); + expected.add("1_[a, c, a, d, a]"); + expected.add("1_[a, c, a, d, a, b]"); + expected.add("1_[a, c, a, d, a, b, r]"); + expected.add("1_[a, c, a, d, a, b, r, a]"); + expected.add("1_[c]"); + expected.add("1_[c, a]"); + expected.add("1_[c, a, d]"); + expected.add("1_[c, a, d, a]"); + expected.add("1_[c, a, d, a, b]"); + expected.add("1_[c, a, d, a, b, r]"); + expected.add("1_[c, a, d, a, b, r, a]"); + expected.add("1_[a, d]"); + expected.add("1_[a, d, a]"); + expected.add("1_[a, d, a, b]"); + expected.add("1_[a, d, a, b, r]"); + expected.add("1_[a, d, a, b, r, a]"); + expected.add("1_[d]"); + expected.add("1_[d, a]"); + expected.add("1_[d, a, b]"); + expected.add("1_[d, a, b, r]"); + expected.add("1_[d, a, b, r, a]"); + + assertEquals(expected.size(), sequences.size()); + + for (String sequence : sequences) { + ListAssert.assertContains(expected, sequence); } - - - @Test - public void testProcessWithTrieProcessor_10() throws Exception { - Trie fixture = new Trie(); - fixture.train(sequence, 11); - - final List sequences = new ArrayList(); - TrieProcessor processor = new TrieProcessor() { - @Override - public TrieProcessor.Result process(List sequence, int count) { - sequences.add(count + "_" + sequence.toString()); - return TrieProcessor.Result.CONTINUE; - } - - }; - - fixture.process(processor); - - List expected = new ArrayList(); - - expected.add("5_[a]"); - expected.add("2_[a, b]"); - expected.add("2_[a, b, r]"); - expected.add("2_[a, b, r, a]"); - expected.add("1_[a, b, r, a, c]"); - expected.add("1_[a, b, r, a, c, a]"); - expected.add("1_[a, b, r, a, c, a, d]"); - expected.add("1_[a, b, r, a, c, a, d, a]"); - expected.add("1_[a, b, r, a, c, a, d, a, b]"); - expected.add("1_[a, b, r, a, c, a, d, a, b, r]"); - expected.add("1_[a, b, r, a, c, a, d, a, b, r, a]"); - expected.add("2_[b]"); - expected.add("2_[b, r]"); - expected.add("2_[b, r, a]"); - expected.add("1_[b, r, a, c]"); - expected.add("1_[b, r, a, c, a]"); - expected.add("1_[b, r, a, c, a, d]"); - expected.add("1_[b, r, a, c, a, d, a]"); - expected.add("1_[b, r, a, c, a, d, a, b]"); - expected.add("1_[b, r, a, c, a, d, a, b, r]"); - expected.add("1_[b, r, a, c, a, d, a, b, r, a]"); - expected.add("2_[r]"); - expected.add("2_[r, a]"); - expected.add("1_[r, a, c]"); - expected.add("1_[r, a, c, a]"); - expected.add("1_[r, a, c, a, d]"); - expected.add("1_[r, a, c, a, d, a]"); - expected.add("1_[r, a, c, a, d, a, b]"); - expected.add("1_[r, a, c, a, d, a, b, r]"); - expected.add("1_[r, a, c, a, d, a, b, r, a]"); - expected.add("1_[a, c]"); - expected.add("1_[a, c, a]"); - expected.add("1_[a, c, a, d]"); - expected.add("1_[a, c, a, d, a]"); - expected.add("1_[a, c, a, d, a, b]"); - expected.add("1_[a, c, a, d, a, b, r]"); - expected.add("1_[a, c, a, d, a, b, r, a]"); - expected.add("1_[c]"); - expected.add("1_[c, a]"); - expected.add("1_[c, a, d]"); - expected.add("1_[c, a, d, a]"); - expected.add("1_[c, a, d, a, b]"); - expected.add("1_[c, a, d, a, b, r]"); - expected.add("1_[c, a, d, a, b, r, a]"); - expected.add("1_[a, d]"); - expected.add("1_[a, d, a]"); - expected.add("1_[a, d, a, b]"); - expected.add("1_[a, d, a, b, r]"); - expected.add("1_[a, d, a, b, r, a]"); - expected.add("1_[d]"); - expected.add("1_[d, a]"); - expected.add("1_[d, a, b]"); - expected.add("1_[d, a, b, r]"); - expected.add("1_[d, a, b, r, a]"); - - assertEquals(expected.size(), sequences.size()); - - for (String sequence : sequences) { - ListAssert.assertContains(expected, sequence); - } - } - - @Test - public void testGetSequencesWithMostOccurrences_1() throws Exception { - Trie fixture = new Trie(); - fixture.train(sequence, 3); - - // get all sequences with a minimal length of one that occur most often - Collection> result = fixture.getSequencesWithOccurrenceCount(1, 0); - - assertEquals(1, result.size()); - - List expected = new ArrayList(); - expected.add("a"); - - ListAssert.assertEquals(expected, result.iterator().next()); - } - - @Test - public void testGetSequencesWithMostOccurrences_2() throws Exception { - Trie fixture = new Trie(); - fixture.train(sequence, 3); - - // get all sequences with a minimal length of one that occur exactly once - Collection> result = fixture.getSequencesWithOccurrenceCount(1, 1); - - assertEquals(11, result.size()); - - List expected = new ArrayList(); - expected.add("r"); - expected.add("a"); - expected.add("c"); - // rac - ListAssert.assertContains((List>) result, expected); - - expected.clear(); - expected.add("a"); - expected.add("c"); - // ac - ListAssert.assertContains((List>) result, expected); - - expected.add("a"); - // aca - ListAssert.assertContains((List>) result, expected); - - expected.clear(); - expected.add("c"); - // c - ListAssert.assertContains((List>) result, expected); - - expected.add("a"); - // ca - ListAssert.assertContains((List>) result, expected); - - expected.add("d"); - // cad - ListAssert.assertContains((List>) result, expected); - - expected.clear(); - expected.add("a"); - expected.add("d"); - // ad - ListAssert.assertContains((List>) result, expected); - - expected.add("a"); - // ada - ListAssert.assertContains((List>) result, expected); - - expected.clear(); - expected.add("d"); - // d - ListAssert.assertContains((List>) result, expected); - - expected.add("a"); - // da - ListAssert.assertContains((List>) result, expected); - - expected.add("b"); - // dab - ListAssert.assertContains((List>) result, expected); - } - - @Test - public void testGetSequencesWithMostOccurrences_3() throws Exception { - Trie fixture = new Trie(); - fixture.train(sequence, 3); - - // get all sequences with a minimal length of one that occur exactly twice - Collection> result = fixture.getSequencesWithOccurrenceCount(1, 2); - - assertEquals(7, result.size()); - - List expected = new ArrayList(); - expected.add("a"); - expected.add("b"); - // ab - ListAssert.assertContains((List>) result, expected); - - expected.add("r"); - // abr - ListAssert.assertContains((List>) result, expected); - - expected.clear(); - expected.add("b"); - // b - ListAssert.assertContains((List>) result, expected); - - expected.add("r"); - // br - ListAssert.assertContains((List>) result, expected); - - expected.add("a"); - // bra - ListAssert.assertContains((List>) result, expected); - - expected.clear(); - expected.add("r"); - // r - ListAssert.assertContains((List>) result, expected); - - expected.add("a"); - // ra - ListAssert.assertContains((List>) result, expected); - } - - @Test - public void testGetSequencesWithMostOccurrences_4() throws Exception { - Trie fixture = new Trie(); - fixture.train(sequence, 3); - - // get all sequences with a minimal length of one that occur exactly three times - Collection> result = fixture.getSequencesWithOccurrenceCount(1, 3); - - assertEquals(0, result.size()); - } - - @Test - public void testGetSequencesWithMostOccurrences_5() throws Exception { - Trie fixture = new Trie(); - fixture.train(sequence, 3); - - // get all sequences with a minimal length of one that occur exactly four times - Collection> result = fixture.getSequencesWithOccurrenceCount(1, 4); - - assertEquals(0, result.size()); - } - - @Test - public void testGetSequencesWithMostOccurrences_6() throws Exception { - Trie fixture = new Trie(); - fixture.train(sequence, 3); - - // get all sequences with a minimal length of one that occur exactly five times - Collection> result = fixture.getSequencesWithOccurrenceCount(1, 5); - - assertEquals(1, result.size()); - - List expected = new ArrayList(); - expected.add("a"); - ListAssert.assertContains((List>) result, expected); - } - - @Test - public void testGetSequencesWithMostOccurrences_7() throws Exception { - Trie fixture = new Trie(); - fixture.train(sequence, 3); - - // get all sequences with a minimal length of two that occur most often - Collection> result = fixture.getSequencesWithOccurrenceCount(2, 0); - - assertEquals(5, result.size()); - - List expected = new ArrayList(); - expected.add("a"); - expected.add("b"); - ListAssert.assertContains((List>) result, expected); - - expected.add("r"); - ListAssert.assertContains((List>) result, expected); - - expected.clear(); - expected.add("b"); - expected.add("r"); - ListAssert.assertContains((List>) result, expected); - - expected.add("a"); - ListAssert.assertContains((List>) result, expected); - - expected.clear(); - expected.add("r"); - expected.add("a"); - ListAssert.assertContains((List>) result, expected); - } - - @Test - public void testGetSequencesWithMostOccurrences_8() throws Exception { - Trie fixture = new Trie(); - fixture.train(sequence, 3); - - // get all sequences with a minimal length of two that occur exactly once - Collection> result = fixture.getSequencesWithOccurrenceCount(2, 1); - - assertEquals(9, result.size()); - - List expected = new ArrayList(); - expected.add("r"); - expected.add("a"); - expected.add("c"); - // rac - ListAssert.assertContains((List>) result, expected); - - expected.clear(); - expected.add("a"); - expected.add("c"); - // ac - ListAssert.assertContains((List>) result, expected); - - expected.add("a"); - // aca - ListAssert.assertContains((List>) result, expected); - - expected.clear(); - expected.add("c"); - expected.add("a"); - // ca - ListAssert.assertContains((List>) result, expected); - - expected.add("d"); - // cad - ListAssert.assertContains((List>) result, expected); - - expected.clear(); - expected.add("a"); - expected.add("d"); - // ad - ListAssert.assertContains((List>) result, expected); - - expected.add("a"); - // ada - ListAssert.assertContains((List>) result, expected); - - expected.clear(); - expected.add("d"); - expected.add("a"); - // da - ListAssert.assertContains((List>) result, expected); - - expected.add("b"); - // dab - ListAssert.assertContains((List>) result, expected); - } - - @Test - public void testGetSequencesWithMostOccurrences_9() throws Exception { - Trie fixture = new Trie(); - fixture.train(sequence, 3); - - // get all sequences with a minimal length of two that occur exactly twice - Collection> result = fixture.getSequencesWithOccurrenceCount(2, 2); - - assertEquals(5, result.size()); - - List expected = new ArrayList(); - expected.add("a"); - expected.add("b"); - // ab - ListAssert.assertContains((List>) result, expected); - - expected.add("r"); - // abr - ListAssert.assertContains((List>) result, expected); - - expected.clear(); - expected.add("b"); - expected.add("r"); - // br - ListAssert.assertContains((List>) result, expected); - - expected.add("a"); - // bra - ListAssert.assertContains((List>) result, expected); - - expected.clear(); - expected.add("r"); - expected.add("a"); - // ra - ListAssert.assertContains((List>) result, expected); - } - - @Test - public void testGetSequencesWithMostOccurrences_10() throws Exception { - Trie fixture = new Trie(); - fixture.train(sequence, 3); - - // get all sequences with a minimal length of two that occur exactly three times - Collection> result = fixture.getSequencesWithOccurrenceCount(2, 3); - - assertEquals(0, result.size()); - } - - @Test - public void testGetSequencesWithMostOccurrences_11() throws Exception { - Trie fixture = new Trie(); - fixture.train(sequence, 3); - - // get all sequences with a minimal length of three that occur most often - Collection> result = fixture.getSequencesWithOccurrenceCount(3, 0); - - assertEquals(2, result.size()); - - List expected = new ArrayList(); - expected.add("a"); - expected.add("b"); - expected.add("r"); - ListAssert.assertContains((List>) result, expected); - - expected.clear(); - expected.add("b"); - expected.add("r"); - expected.add("a"); - ListAssert.assertContains((List>) result, expected); - } - - @Test - public void testGetSequencesWithMostOccurrences_12() throws Exception { - Trie fixture = new Trie(); - fixture.train(sequence, 3); - - // get all sequences with a minimal length of three that occur exactly once - Collection> result = fixture.getSequencesWithOccurrenceCount(3, 1); - - assertEquals(5, result.size()); - - List expected = new ArrayList(); - expected.add("r"); - expected.add("a"); - expected.add("c"); - // rac - ListAssert.assertContains((List>) result, expected); - - expected.clear(); - expected.add("a"); - expected.add("c"); - expected.add("a"); - // aca - ListAssert.assertContains((List>) result, expected); - - expected.clear(); - expected.add("c"); - expected.add("a"); - expected.add("d"); - // cad - ListAssert.assertContains((List>) result, expected); - - expected.clear(); - expected.add("a"); - expected.add("d"); - expected.add("a"); - // ada - ListAssert.assertContains((List>) result, expected); - - expected.clear(); - expected.add("d"); - expected.add("a"); - expected.add("b"); - // dab - ListAssert.assertContains((List>) result, expected); - } - - @Test - public void testGetSequencesWithMostOccurrences_13() throws Exception { - Trie fixture = new Trie(); - fixture.train(sequence, 3); - - // get all sequences with a minimal length of four that occur most often - Collection> result = fixture.getSequencesWithOccurrenceCount(4, 0); - - // none of these exist, as the tree is only trained with sequences of length 3 - assertEquals(0, result.size()); - } - - @Test - public void testGetSequencesWithMostOccurrences_14() throws Exception { - Trie fixture = new Trie(); - fixture.train(sequence, 3); - - // get all sequences with a minimal length of four that occur exactly once - Collection> result = fixture.getSequencesWithOccurrenceCount(4, 1); - - // none of these exist, as the tree is only trained with sequences of length 3 - assertEquals(0, result.size()); - } - - @Test - public void testGetCount_1() throws Exception { - Trie fixture = new Trie(); - fixture.train(sequence, 3); - List subSequence = new ArrayList(); - subSequence.add("a"); - - int result = fixture.getCount(subSequence); - - assertEquals(5, result); - } - - @Test - public void testGetCount_2() throws Exception { - Trie fixture = new Trie(); - fixture.train(sequence, 3); - List subSequence = new ArrayList(); - subSequence.add("a"); - subSequence.add("b"); - - int result = fixture.getCount(subSequence); - - assertEquals(2, result); - } - - @Test - public void testGetCount_3() throws Exception { - Trie fixture = new Trie(); - fixture.train(sequence, 3); - List subSequence = new ArrayList(); - subSequence.add("x"); - - int result = fixture.getCount(subSequence); - - assertEquals(0, result); - } - - @Test - public void testGetCount_4() throws Exception { - Trie fixture = new Trie(); - fixture.train(sequence, 3); - List subSequence = new ArrayList(); - - int result = fixture.getCount(subSequence, "a"); - - assertEquals(5, result); - } - - @Test - public void testGetCount_5() throws Exception { - Trie fixture = new Trie(); - fixture.train(sequence, 3); - List subSequence = new ArrayList(); - subSequence.add("a"); - subSequence.add("b"); - - int result = fixture.getCount(subSequence, "r"); - - assertEquals(2, result); - } - - @Test - public void testGetCount_6() throws Exception { - Trie fixture = new Trie(); - fixture.train(sequence, 3); - List subSequence = new ArrayList(); - - int result = fixture.getCount(subSequence, "x"); - - assertEquals(0, result); - } - - @Test - public void testGetFollowingSymbols_1() throws Exception { - Trie fixture = new Trie(); - fixture.train(sequence, 3); - List subSequence = new ArrayList(); - subSequence.add("a"); - Collection expected = new ArrayList(); - expected.add("b"); - expected.add("c"); - expected.add("d"); - - Collection result = fixture.getFollowingSymbols(subSequence); - - assertCollectionContent(expected, result); - } - - @Test - public void testGetFollowingSymbols_2() throws Exception { - Trie fixture = new Trie(); - fixture.train(sequence, 3); - List subSequence = new ArrayList(); - subSequence.add("a"); - subSequence.add("b"); - subSequence.add("r"); - - Collection result = fixture.getFollowingSymbols(subSequence); - - assertEquals(0, result.size()); - } - - @Test - public void testGetFollowingSymbols_3() throws Exception { - Trie fixture = new Trie(); - fixture.train(sequence, 3); - List subSequence = new ArrayList(); - subSequence.add("x"); - - Collection result = fixture.getFollowingSymbols(subSequence); - - assertEquals(0, result.size()); - } - - @Test - public void testGetNumLeafAncestors_1() throws Exception { - Trie fixture = new Trie(); - fixture.train(sequence, 3); - - int result = fixture.getNumLeafAncestors(); - - assertEquals(7, result); - } - - @Test - public void testGetNumLeafs_1() throws Exception { - Trie fixture = new Trie(); - fixture.train(sequence, 3); - - int result = fixture.getNumLeafs(); - - assertEquals(7, result); - } - - @Test - public void testGetNumSymbols_1() throws Exception { - Trie fixture = new Trie(); - fixture.train(sequence, 3); - - int result = fixture.getNumSymbols(); - - assertEquals(5, result); - } - - @Test - public void testTrain_1() throws Exception { - Trie fixture = new Trie(); - int maxOrder = 3; - - fixture.train(sequence, maxOrder); - - // check if symbols are correct - assertCollectionContent(symbols, fixture.getKnownSymbols()); - - // check if counters are correct and only the correct nodes exist - TrieNode root = fixture.find(new ArrayList()); - TrieNode root_a = root.getChild("a"); - TrieNode root_a_a = root_a.getChild("a"); - TrieNode root_a_b = root_a.getChild("b"); - TrieNode root_a_b_a = root_a_b.getChild("a"); - TrieNode root_a_b_b = root_a_b.getChild("b"); - TrieNode root_a_b_c = root_a_b.getChild("c"); - TrieNode root_a_b_d = root_a_b.getChild("d"); - TrieNode root_a_b_r = root_a_b.getChild("r"); - TrieNode root_a_c = root_a.getChild("c"); - TrieNode root_a_c_a = root_a_c.getChild("a"); - TrieNode root_a_c_b = root_a_c.getChild("b"); - TrieNode root_a_c_c = root_a_c.getChild("c"); - TrieNode root_a_c_d = root_a_c.getChild("d"); - TrieNode root_a_c_r = root_a_c.getChild("r"); - TrieNode root_a_d = root_a.getChild("d"); - TrieNode root_a_d_a = root_a_d.getChild("a"); - TrieNode root_a_d_b = root_a_d.getChild("b"); - TrieNode root_a_d_c = root_a_d.getChild("c"); - TrieNode root_a_d_d = root_a_d.getChild("d"); - TrieNode root_a_d_r = root_a_d.getChild("r"); - TrieNode root_a_r = root_a.getChild("r"); - TrieNode root_b = root.getChild("b"); - TrieNode root_b_a = root_b.getChild("a"); - TrieNode root_b_b = root_b.getChild("b"); - TrieNode root_b_c = root_b.getChild("c"); - TrieNode root_b_d = root_b.getChild("d"); - TrieNode root_b_r = root_b.getChild("r"); - TrieNode root_b_r_a = root_b_r.getChild("a"); - TrieNode root_b_r_b = root_b_r.getChild("b"); - TrieNode root_b_r_c = root_b_r.getChild("c"); - TrieNode root_b_r_d = root_b_r.getChild("d"); - TrieNode root_b_r_r = root_b_r.getChild("r"); - TrieNode root_c = root.getChild("c"); - TrieNode root_c_a = root_c.getChild("a"); - TrieNode root_c_a_a = root_c_a.getChild("a"); - TrieNode root_c_a_b = root_c_a.getChild("b"); - TrieNode root_c_a_c = root_c_a.getChild("c"); - TrieNode root_c_a_d = root_c_a.getChild("d"); - TrieNode root_c_a_r = root_c_a.getChild("r"); - TrieNode root_c_b = root_c.getChild("b"); - TrieNode root_c_c = root_c.getChild("c"); - TrieNode root_c_d = root_c.getChild("d"); - TrieNode root_c_r = root_c.getChild("r"); - TrieNode root_d = root.getChild("d"); - TrieNode root_d_a = root_d.getChild("a"); - TrieNode root_d_a_a = root_d_a.getChild("a"); - TrieNode root_d_a_b = root_d_a.getChild("b"); - TrieNode root_d_a_c = root_d_a.getChild("c"); - TrieNode root_d_a_d = root_d_a.getChild("d"); - TrieNode root_d_a_r = root_d_a.getChild("r"); - TrieNode root_d_b = root_d.getChild("b"); - TrieNode root_d_c = root_d.getChild("c"); - TrieNode root_d_d = root_d.getChild("d"); - TrieNode root_d_r = root_d.getChild("r"); - TrieNode root_r = root.getChild("r"); - TrieNode root_r_a = root_r.getChild("a"); - TrieNode root_r_a_a = root_r_a.getChild("a"); - TrieNode root_r_a_b = root_r_a.getChild("b"); - TrieNode root_r_a_c = root_r_a.getChild("c"); - TrieNode root_r_a_d = root_r_a.getChild("d"); - TrieNode root_r_a_r = root_r_a.getChild("r"); - TrieNode root_r_b = root_r.getChild("b"); - TrieNode root_r_c = root_r.getChild("c"); - TrieNode root_r_d = root_r.getChild("d"); - TrieNode root_r_r = root_r.getChild("r"); - - assertEquals(5, root_a.getCount()); - assertNull(root_a_a); - assertEquals(2, root_a_b.getCount()); - assertNull(root_a_b_a); - assertNull(root_a_b_b); - assertNull(root_a_b_c); - assertNull(root_a_b_d); - assertEquals(2, root_a_b_r.getCount()); - assertEquals(1, root_a_c.getCount()); - assertEquals(1, root_a_c_a.getCount()); - assertNull(root_a_c_b); - assertNull(root_a_c_c); - assertNull(root_a_c_d); - assertNull(root_a_c_r); - assertEquals(1, root_a_d.getCount()); - assertEquals(1, root_a_d_a.getCount()); - assertNull(root_a_d_b); - assertNull(root_a_d_c); - assertNull(root_a_d_d); - assertNull(root_a_d_r); - assertNull(root_a_r); - - assertEquals(2, root_b.getCount()); - assertNull(root_b_a); - assertNull(root_b_b); - assertNull(root_b_c); - assertNull(root_b_d); - assertEquals(2, root_b_r.getCount()); - assertEquals(2, root_b_r_a.getCount()); - assertNull(root_b_r_b); - assertNull(root_b_r_c); - assertNull(root_b_r_d); - assertNull(root_b_r_r); - - assertEquals(1, root_c.getCount()); - assertEquals(1, root_c_a.getCount()); - assertNull(root_c_a_a); - assertNull(root_c_a_b); - assertNull(root_c_a_c); - assertEquals(1, root_c_a_d.getCount()); - assertNull(root_c_a_r); - assertNull(root_c_b); - assertNull(root_c_c); - assertNull(root_c_d); - assertNull(root_c_r); - - assertEquals(1, root_d.getCount()); - assertEquals(1, root_d_a.getCount()); - assertNull(root_d_a_a); - assertEquals(1, root_d_a_b.getCount()); - assertNull(root_d_a_c); - assertNull(root_d_a_d); - assertNull(root_d_a_r); - assertNull(root_d_b); - assertNull(root_d_c); - assertNull(root_d_d); - assertNull(root_d_r); - - assertEquals(2, root_r.getCount()); - assertEquals(2, root_r_a.getCount()); - assertNull(root_r_a_a); - assertNull(root_r_a_b); - assertEquals(1, root_r_a_c.getCount()); - assertNull(root_r_a_d); - assertNull(root_r_a_r); - assertNull(root_r_b); - assertNull(root_r_c); - assertNull(root_r_d); - assertNull(root_r_r); - - // check if leafs are really leafs - assertTrue(root_a_b_r.isLeaf()); - assertTrue(root_a_c_a.isLeaf()); - assertTrue(root_a_d_a.isLeaf()); - assertTrue(root_b_r_a.isLeaf()); - assertTrue(root_c_a_d.isLeaf()); - assertTrue(root_d_a_b.isLeaf()); - assertTrue(root_r_a_c.isLeaf()); - } - - @Test - public void testTrain_2() throws Exception { - Trie fixture = new Trie(); - int maxOrder = 0; - - fixture.train(sequence, maxOrder); - - assertTrue(fixture.getKnownSymbols().isEmpty()); - } - - @Test - public void testTrain_3() throws Exception { - Trie fixture = new Trie(); - List sequence = new ArrayList(); - int maxOrder = 1; - - fixture.train(sequence, maxOrder); - - assertTrue(fixture.getKnownSymbols().isEmpty()); - } - - @Test - public void testTrain_4() throws Exception { - Trie fixture = new Trie(); - List sequence = new ArrayList(); - sequence.add("a"); - sequence.add("b"); - int maxOrder = 3; - - fixture.train(sequence, maxOrder); - - assertCollectionContent(sequence, fixture.getKnownSymbols()); - TrieNode root = fixture.find(new ArrayList()); - TrieNode root_a = root.getChild("a"); - TrieNode root_a_a = root_a.getChild("a"); - TrieNode root_a_b = root_a.getChild("b"); - TrieNode root_b = root.getChild("b"); - TrieNode root_b_a = root_b.getChild("a"); - TrieNode root_b_b = root_b.getChild("b"); - - assertEquals(1, root_a.getCount()); - assertNull(root_a_a); - assertEquals(1, root_a_b.getCount()); - assertEquals(1, root_b.getCount()); - assertNull(root_b_a); - assertNull(root_b_b); - - assertTrue(root_a_b.isLeaf()); - assertTrue(root_b.isLeaf()); - } - - @Test - public void testEdgeEdge_1() throws Exception { - Edge result = new Edge(); - - assertNotNull(result); - } - - @Test - public void testTrieVertexTrieVertex_1() throws Exception { - String id = "idString"; - - TrieVertex result = new TrieVertex(id); - - assertNotNull(result); - } - - @Test - public void testTrieVertexToString_1() throws Exception { - String id = "idString"; - TrieVertex fixture = new TrieVertex(id); - - String result = fixture.toString(); - - assertEquals(id, result); - } - - @Test - public void testEquals_1() throws Exception { - Trie trieOther = new Trie(); - Trie fixture = new Trie(); - - boolean result = fixture.equals(trieOther); - - assertEquals(true, result); - } - - @Test - public void testEquals_2() throws Exception { - Trie trieOther = new Trie(); - trieOther.train(sequence, 2); - Trie fixture = new Trie(); - fixture.train(sequence, 2); - - boolean result = fixture.equals(trieOther); - - assertEquals(true, result); - } - - @Test - public void testEquals_3() throws Exception { - Trie trieOther = new Trie(); - trieOther.train(sequence, 2); - Trie fixture = new Trie(); - fixture.train(sequence, 3); - - boolean result = fixture.equals(trieOther); - - assertEquals(false, result); - } - - @Test - public void testEquals_4() throws Exception { - Trie trieOther = new Trie(); - Trie fixture = new Trie(); - fixture.train(sequence, 2); - - boolean result = fixture.equals(trieOther); - - assertEquals(false, result); - } - - @Test - public void testEquals_5() throws Exception { - Trie trieOther = new Trie(); - trieOther.train(sequence, 2); - Trie fixture = new Trie(); - - boolean result = fixture.equals(trieOther); - - assertEquals(false, result); - } - - @Test - public void testEquals_6() throws Exception { - Trie fixture = new Trie(); - fixture.train(sequence, 2); - - boolean result = fixture.equals(fixture); - - assertEquals(true, result); - } - - @Test - public void testEquals_7() throws Exception { - Trie fixture = new Trie(); - fixture.train(sequence, 2); - - boolean result = fixture.equals(null); - - assertEquals(false, result); - } - - @Before - public void setUp() throws Exception { - sequence = new ArrayList(); - sequence.add("a"); - sequence.add("b"); - sequence.add("r"); - sequence.add("a"); - sequence.add("c"); - sequence.add("a"); - sequence.add("d"); - sequence.add("a"); - sequence.add("b"); - sequence.add("r"); - sequence.add("a"); - - symbols = new HashSet(); - symbols.add("a"); - symbols.add("b"); - symbols.add("c"); - symbols.add("d"); - symbols.add("r"); - } - - public static void main(String[] args) { - new org.junit.runner.JUnitCore().run(TrieTest.class); - } + } + + @Test + public void testGetSequencesWithMostOccurrences_1() throws Exception { + Trie fixture = new Trie(); + fixture.train(sequence, 3); + + // get all sequences with a minimal length of one that occur most often + Collection> result = fixture.getSequencesWithOccurrenceCount(1, 0); + + assertEquals(1, result.size()); + + List expected = new ArrayList(); + expected.add("a"); + + ListAssert.assertEquals(expected, result.iterator().next()); + } + + @Test + public void testGetSequencesWithMostOccurrences_2() throws Exception { + Trie fixture = new Trie(); + fixture.train(sequence, 3); + + // get all sequences with a minimal length of one that occur exactly once + Collection> result = fixture.getSequencesWithOccurrenceCount(1, 1); + + assertEquals(11, result.size()); + + List expected = new ArrayList(); + expected.add("r"); + expected.add("a"); + expected.add("c"); + // rac + ListAssert.assertContains((List>) result, expected); + + expected.clear(); + expected.add("a"); + expected.add("c"); + // ac + ListAssert.assertContains((List>) result, expected); + + expected.add("a"); + // aca + ListAssert.assertContains((List>) result, expected); + + expected.clear(); + expected.add("c"); + // c + ListAssert.assertContains((List>) result, expected); + + expected.add("a"); + // ca + ListAssert.assertContains((List>) result, expected); + + expected.add("d"); + // cad + ListAssert.assertContains((List>) result, expected); + + expected.clear(); + expected.add("a"); + expected.add("d"); + // ad + ListAssert.assertContains((List>) result, expected); + + expected.add("a"); + // ada + ListAssert.assertContains((List>) result, expected); + + expected.clear(); + expected.add("d"); + // d + ListAssert.assertContains((List>) result, expected); + + expected.add("a"); + // da + ListAssert.assertContains((List>) result, expected); + + expected.add("b"); + // dab + ListAssert.assertContains((List>) result, expected); + } + + @Test + public void testGetSequencesWithMostOccurrences_3() throws Exception { + Trie fixture = new Trie(); + fixture.train(sequence, 3); + + // get all sequences with a minimal length of one that occur exactly twice + Collection> result = fixture.getSequencesWithOccurrenceCount(1, 2); + + assertEquals(7, result.size()); + + List expected = new ArrayList(); + expected.add("a"); + expected.add("b"); + // ab + ListAssert.assertContains((List>) result, expected); + + expected.add("r"); + // abr + ListAssert.assertContains((List>) result, expected); + + expected.clear(); + expected.add("b"); + // b + ListAssert.assertContains((List>) result, expected); + + expected.add("r"); + // br + ListAssert.assertContains((List>) result, expected); + + expected.add("a"); + // bra + ListAssert.assertContains((List>) result, expected); + + expected.clear(); + expected.add("r"); + // r + ListAssert.assertContains((List>) result, expected); + + expected.add("a"); + // ra + ListAssert.assertContains((List>) result, expected); + } + + @Test + public void testGetSequencesWithMostOccurrences_4() throws Exception { + Trie fixture = new Trie(); + fixture.train(sequence, 3); + + // get all sequences with a minimal length of one that occur exactly three times + Collection> result = fixture.getSequencesWithOccurrenceCount(1, 3); + + assertEquals(0, result.size()); + } + + @Test + public void testGetSequencesWithMostOccurrences_5() throws Exception { + Trie fixture = new Trie(); + fixture.train(sequence, 3); + + // get all sequences with a minimal length of one that occur exactly four times + Collection> result = fixture.getSequencesWithOccurrenceCount(1, 4); + + assertEquals(0, result.size()); + } + + @Test + public void testGetSequencesWithMostOccurrences_6() throws Exception { + Trie fixture = new Trie(); + fixture.train(sequence, 3); + + // get all sequences with a minimal length of one that occur exactly five times + Collection> result = fixture.getSequencesWithOccurrenceCount(1, 5); + + assertEquals(1, result.size()); + + List expected = new ArrayList(); + expected.add("a"); + ListAssert.assertContains((List>) result, expected); + } + + @Test + public void testGetSequencesWithMostOccurrences_7() throws Exception { + Trie fixture = new Trie(); + fixture.train(sequence, 3); + + // get all sequences with a minimal length of two that occur most often + Collection> result = fixture.getSequencesWithOccurrenceCount(2, 0); + + assertEquals(5, result.size()); + + List expected = new ArrayList(); + expected.add("a"); + expected.add("b"); + ListAssert.assertContains((List>) result, expected); + + expected.add("r"); + ListAssert.assertContains((List>) result, expected); + + expected.clear(); + expected.add("b"); + expected.add("r"); + ListAssert.assertContains((List>) result, expected); + + expected.add("a"); + ListAssert.assertContains((List>) result, expected); + + expected.clear(); + expected.add("r"); + expected.add("a"); + ListAssert.assertContains((List>) result, expected); + } + + @Test + public void testGetSequencesWithMostOccurrences_8() throws Exception { + Trie fixture = new Trie(); + fixture.train(sequence, 3); + + // get all sequences with a minimal length of two that occur exactly once + Collection> result = fixture.getSequencesWithOccurrenceCount(2, 1); + + assertEquals(9, result.size()); + + List expected = new ArrayList(); + expected.add("r"); + expected.add("a"); + expected.add("c"); + // rac + ListAssert.assertContains((List>) result, expected); + + expected.clear(); + expected.add("a"); + expected.add("c"); + // ac + ListAssert.assertContains((List>) result, expected); + + expected.add("a"); + // aca + ListAssert.assertContains((List>) result, expected); + + expected.clear(); + expected.add("c"); + expected.add("a"); + // ca + ListAssert.assertContains((List>) result, expected); + + expected.add("d"); + // cad + ListAssert.assertContains((List>) result, expected); + + expected.clear(); + expected.add("a"); + expected.add("d"); + // ad + ListAssert.assertContains((List>) result, expected); + + expected.add("a"); + // ada + ListAssert.assertContains((List>) result, expected); + + expected.clear(); + expected.add("d"); + expected.add("a"); + // da + ListAssert.assertContains((List>) result, expected); + + expected.add("b"); + // dab + ListAssert.assertContains((List>) result, expected); + } + + @Test + public void testGetSequencesWithMostOccurrences_9() throws Exception { + Trie fixture = new Trie(); + fixture.train(sequence, 3); + + // get all sequences with a minimal length of two that occur exactly twice + Collection> result = fixture.getSequencesWithOccurrenceCount(2, 2); + + assertEquals(5, result.size()); + + List expected = new ArrayList(); + expected.add("a"); + expected.add("b"); + // ab + ListAssert.assertContains((List>) result, expected); + + expected.add("r"); + // abr + ListAssert.assertContains((List>) result, expected); + + expected.clear(); + expected.add("b"); + expected.add("r"); + // br + ListAssert.assertContains((List>) result, expected); + + expected.add("a"); + // bra + ListAssert.assertContains((List>) result, expected); + + expected.clear(); + expected.add("r"); + expected.add("a"); + // ra + ListAssert.assertContains((List>) result, expected); + } + + @Test + public void testGetSequencesWithMostOccurrences_10() throws Exception { + Trie fixture = new Trie(); + fixture.train(sequence, 3); + + // get all sequences with a minimal length of two that occur exactly three times + Collection> result = fixture.getSequencesWithOccurrenceCount(2, 3); + + assertEquals(0, result.size()); + } + + @Test + public void testGetSequencesWithMostOccurrences_11() throws Exception { + Trie fixture = new Trie(); + fixture.train(sequence, 3); + + // get all sequences with a minimal length of three that occur most often + Collection> result = fixture.getSequencesWithOccurrenceCount(3, 0); + + assertEquals(2, result.size()); + + List expected = new ArrayList(); + expected.add("a"); + expected.add("b"); + expected.add("r"); + ListAssert.assertContains((List>) result, expected); + + expected.clear(); + expected.add("b"); + expected.add("r"); + expected.add("a"); + ListAssert.assertContains((List>) result, expected); + } + + @Test + public void testGetSequencesWithMostOccurrences_12() throws Exception { + Trie fixture = new Trie(); + fixture.train(sequence, 3); + + // get all sequences with a minimal length of three that occur exactly once + Collection> result = fixture.getSequencesWithOccurrenceCount(3, 1); + + assertEquals(5, result.size()); + + List expected = new ArrayList(); + expected.add("r"); + expected.add("a"); + expected.add("c"); + // rac + ListAssert.assertContains((List>) result, expected); + + expected.clear(); + expected.add("a"); + expected.add("c"); + expected.add("a"); + // aca + ListAssert.assertContains((List>) result, expected); + + expected.clear(); + expected.add("c"); + expected.add("a"); + expected.add("d"); + // cad + ListAssert.assertContains((List>) result, expected); + + expected.clear(); + expected.add("a"); + expected.add("d"); + expected.add("a"); + // ada + ListAssert.assertContains((List>) result, expected); + + expected.clear(); + expected.add("d"); + expected.add("a"); + expected.add("b"); + // dab + ListAssert.assertContains((List>) result, expected); + } + + @Test + public void testGetSequencesWithMostOccurrences_13() throws Exception { + Trie fixture = new Trie(); + fixture.train(sequence, 3); + + // get all sequences with a minimal length of four that occur most often + Collection> result = fixture.getSequencesWithOccurrenceCount(4, 0); + + // none of these exist, as the tree is only trained with sequences of length 3 + assertEquals(0, result.size()); + } + + @Test + public void testGetSequencesWithMostOccurrences_14() throws Exception { + Trie fixture = new Trie(); + fixture.train(sequence, 3); + + // get all sequences with a minimal length of four that occur exactly once + Collection> result = fixture.getSequencesWithOccurrenceCount(4, 1); + + // none of these exist, as the tree is only trained with sequences of length 3 + assertEquals(0, result.size()); + } + + @Test + public void testGetCount_1() throws Exception { + Trie fixture = new Trie(); + fixture.train(sequence, 3); + List subSequence = new ArrayList(); + subSequence.add("a"); + + int result = fixture.getCount(subSequence); + + assertEquals(5, result); + } + + @Test + public void testGetCount_2() throws Exception { + Trie fixture = new Trie(); + fixture.train(sequence, 3); + List subSequence = new ArrayList(); + subSequence.add("a"); + subSequence.add("b"); + + int result = fixture.getCount(subSequence); + + assertEquals(2, result); + } + + @Test + public void testGetCount_3() throws Exception { + Trie fixture = new Trie(); + fixture.train(sequence, 3); + List subSequence = new ArrayList(); + subSequence.add("x"); + + int result = fixture.getCount(subSequence); + + assertEquals(0, result); + } + + @Test + public void testGetCount_4() throws Exception { + Trie fixture = new Trie(); + fixture.train(sequence, 3); + List subSequence = new ArrayList(); + + int result = fixture.getCount(subSequence, "a"); + + assertEquals(5, result); + } + + @Test + public void testGetCount_5() throws Exception { + Trie fixture = new Trie(); + fixture.train(sequence, 3); + List subSequence = new ArrayList(); + subSequence.add("a"); + subSequence.add("b"); + + int result = fixture.getCount(subSequence, "r"); + + assertEquals(2, result); + } + + @Test + public void testGetCount_6() throws Exception { + Trie fixture = new Trie(); + fixture.train(sequence, 3); + List subSequence = new ArrayList(); + + int result = fixture.getCount(subSequence, "x"); + + assertEquals(0, result); + } + + @Test + public void testGetFollowingSymbols_1() throws Exception { + Trie fixture = new Trie(); + fixture.train(sequence, 3); + List subSequence = new ArrayList(); + subSequence.add("a"); + Collection expected = new ArrayList(); + expected.add("b"); + expected.add("c"); + expected.add("d"); + + Collection result = fixture.getFollowingSymbols(subSequence); + + assertCollectionContent(expected, result); + } + + @Test + public void testGetFollowingSymbols_2() throws Exception { + Trie fixture = new Trie(); + fixture.train(sequence, 3); + List subSequence = new ArrayList(); + subSequence.add("a"); + subSequence.add("b"); + subSequence.add("r"); + + Collection result = fixture.getFollowingSymbols(subSequence); + + assertEquals(0, result.size()); + } + + @Test + public void testGetFollowingSymbols_3() throws Exception { + Trie fixture = new Trie(); + fixture.train(sequence, 3); + List subSequence = new ArrayList(); + subSequence.add("x"); + + Collection result = fixture.getFollowingSymbols(subSequence); + + assertEquals(0, result.size()); + } + + @Test + public void testGetNumLeafAncestors_1() throws Exception { + Trie fixture = new Trie(); + fixture.train(sequence, 3); + + int result = fixture.getNumLeafAncestors(); + + assertEquals(7, result); + } + + @Test + public void testGetNumLeafs_1() throws Exception { + Trie fixture = new Trie(); + fixture.train(sequence, 3); + + int result = fixture.getNumLeafs(); + + assertEquals(7, result); + } + + @Test + public void testGetNumSymbols_1() throws Exception { + Trie fixture = new Trie(); + fixture.train(sequence, 3); + + int result = fixture.getNumSymbols(); + + assertEquals(5, result); + } + + @Test + public void testTrain_1() throws Exception { + Trie fixture = new Trie(); + int maxOrder = 3; + + fixture.train(sequence, maxOrder); + + // check if symbols are correct + assertCollectionContent(symbols, fixture.getKnownSymbols()); + + // check if counters are correct and only the correct nodes exist + TrieNode root = fixture.find(new ArrayList()); + TrieNode root_a = root.getChild("a"); + TrieNode root_a_a = root_a.getChild("a"); + TrieNode root_a_b = root_a.getChild("b"); + TrieNode root_a_b_a = root_a_b.getChild("a"); + TrieNode root_a_b_b = root_a_b.getChild("b"); + TrieNode root_a_b_c = root_a_b.getChild("c"); + TrieNode root_a_b_d = root_a_b.getChild("d"); + TrieNode root_a_b_r = root_a_b.getChild("r"); + TrieNode root_a_c = root_a.getChild("c"); + TrieNode root_a_c_a = root_a_c.getChild("a"); + TrieNode root_a_c_b = root_a_c.getChild("b"); + TrieNode root_a_c_c = root_a_c.getChild("c"); + TrieNode root_a_c_d = root_a_c.getChild("d"); + TrieNode root_a_c_r = root_a_c.getChild("r"); + TrieNode root_a_d = root_a.getChild("d"); + TrieNode root_a_d_a = root_a_d.getChild("a"); + TrieNode root_a_d_b = root_a_d.getChild("b"); + TrieNode root_a_d_c = root_a_d.getChild("c"); + TrieNode root_a_d_d = root_a_d.getChild("d"); + TrieNode root_a_d_r = root_a_d.getChild("r"); + TrieNode root_a_r = root_a.getChild("r"); + TrieNode root_b = root.getChild("b"); + TrieNode root_b_a = root_b.getChild("a"); + TrieNode root_b_b = root_b.getChild("b"); + TrieNode root_b_c = root_b.getChild("c"); + TrieNode root_b_d = root_b.getChild("d"); + TrieNode root_b_r = root_b.getChild("r"); + TrieNode root_b_r_a = root_b_r.getChild("a"); + TrieNode root_b_r_b = root_b_r.getChild("b"); + TrieNode root_b_r_c = root_b_r.getChild("c"); + TrieNode root_b_r_d = root_b_r.getChild("d"); + TrieNode root_b_r_r = root_b_r.getChild("r"); + TrieNode root_c = root.getChild("c"); + TrieNode root_c_a = root_c.getChild("a"); + TrieNode root_c_a_a = root_c_a.getChild("a"); + TrieNode root_c_a_b = root_c_a.getChild("b"); + TrieNode root_c_a_c = root_c_a.getChild("c"); + TrieNode root_c_a_d = root_c_a.getChild("d"); + TrieNode root_c_a_r = root_c_a.getChild("r"); + TrieNode root_c_b = root_c.getChild("b"); + TrieNode root_c_c = root_c.getChild("c"); + TrieNode root_c_d = root_c.getChild("d"); + TrieNode root_c_r = root_c.getChild("r"); + TrieNode root_d = root.getChild("d"); + TrieNode root_d_a = root_d.getChild("a"); + TrieNode root_d_a_a = root_d_a.getChild("a"); + TrieNode root_d_a_b = root_d_a.getChild("b"); + TrieNode root_d_a_c = root_d_a.getChild("c"); + TrieNode root_d_a_d = root_d_a.getChild("d"); + TrieNode root_d_a_r = root_d_a.getChild("r"); + TrieNode root_d_b = root_d.getChild("b"); + TrieNode root_d_c = root_d.getChild("c"); + TrieNode root_d_d = root_d.getChild("d"); + TrieNode root_d_r = root_d.getChild("r"); + TrieNode root_r = root.getChild("r"); + TrieNode root_r_a = root_r.getChild("a"); + TrieNode root_r_a_a = root_r_a.getChild("a"); + TrieNode root_r_a_b = root_r_a.getChild("b"); + TrieNode root_r_a_c = root_r_a.getChild("c"); + TrieNode root_r_a_d = root_r_a.getChild("d"); + TrieNode root_r_a_r = root_r_a.getChild("r"); + TrieNode root_r_b = root_r.getChild("b"); + TrieNode root_r_c = root_r.getChild("c"); + TrieNode root_r_d = root_r.getChild("d"); + TrieNode root_r_r = root_r.getChild("r"); + + assertEquals(5, root_a.getCount()); + assertNull(root_a_a); + assertEquals(2, root_a_b.getCount()); + assertNull(root_a_b_a); + assertNull(root_a_b_b); + assertNull(root_a_b_c); + assertNull(root_a_b_d); + assertEquals(2, root_a_b_r.getCount()); + assertEquals(1, root_a_c.getCount()); + assertEquals(1, root_a_c_a.getCount()); + assertNull(root_a_c_b); + assertNull(root_a_c_c); + assertNull(root_a_c_d); + assertNull(root_a_c_r); + assertEquals(1, root_a_d.getCount()); + assertEquals(1, root_a_d_a.getCount()); + assertNull(root_a_d_b); + assertNull(root_a_d_c); + assertNull(root_a_d_d); + assertNull(root_a_d_r); + assertNull(root_a_r); + + assertEquals(2, root_b.getCount()); + assertNull(root_b_a); + assertNull(root_b_b); + assertNull(root_b_c); + assertNull(root_b_d); + assertEquals(2, root_b_r.getCount()); + assertEquals(2, root_b_r_a.getCount()); + assertNull(root_b_r_b); + assertNull(root_b_r_c); + assertNull(root_b_r_d); + assertNull(root_b_r_r); + + assertEquals(1, root_c.getCount()); + assertEquals(1, root_c_a.getCount()); + assertNull(root_c_a_a); + assertNull(root_c_a_b); + assertNull(root_c_a_c); + assertEquals(1, root_c_a_d.getCount()); + assertNull(root_c_a_r); + assertNull(root_c_b); + assertNull(root_c_c); + assertNull(root_c_d); + assertNull(root_c_r); + + assertEquals(1, root_d.getCount()); + assertEquals(1, root_d_a.getCount()); + assertNull(root_d_a_a); + assertEquals(1, root_d_a_b.getCount()); + assertNull(root_d_a_c); + assertNull(root_d_a_d); + assertNull(root_d_a_r); + assertNull(root_d_b); + assertNull(root_d_c); + assertNull(root_d_d); + assertNull(root_d_r); + + assertEquals(2, root_r.getCount()); + assertEquals(2, root_r_a.getCount()); + assertNull(root_r_a_a); + assertNull(root_r_a_b); + assertEquals(1, root_r_a_c.getCount()); + assertNull(root_r_a_d); + assertNull(root_r_a_r); + assertNull(root_r_b); + assertNull(root_r_c); + assertNull(root_r_d); + assertNull(root_r_r); + + // check if leafs are really leafs + assertTrue(root_a_b_r.isLeaf()); + assertTrue(root_a_c_a.isLeaf()); + assertTrue(root_a_d_a.isLeaf()); + assertTrue(root_b_r_a.isLeaf()); + assertTrue(root_c_a_d.isLeaf()); + assertTrue(root_d_a_b.isLeaf()); + assertTrue(root_r_a_c.isLeaf()); + } + + @Test + public void testTrain_2() throws Exception { + Trie fixture = new Trie(); + int maxOrder = 0; + + fixture.train(sequence, maxOrder); + + assertTrue(fixture.getKnownSymbols().isEmpty()); + } + + @Test + public void testTrain_3() throws Exception { + Trie fixture = new Trie(); + List sequence = new ArrayList(); + int maxOrder = 1; + + fixture.train(sequence, maxOrder); + + assertTrue(fixture.getKnownSymbols().isEmpty()); + } + + @Test + public void testTrain_4() throws Exception { + Trie fixture = new Trie(); + List sequence = new ArrayList(); + sequence.add("a"); + sequence.add("b"); + int maxOrder = 3; + + fixture.train(sequence, maxOrder); + + assertCollectionContent(sequence, fixture.getKnownSymbols()); + TrieNode root = fixture.find(new ArrayList()); + TrieNode root_a = root.getChild("a"); + TrieNode root_a_a = root_a.getChild("a"); + TrieNode root_a_b = root_a.getChild("b"); + TrieNode root_b = root.getChild("b"); + TrieNode root_b_a = root_b.getChild("a"); + TrieNode root_b_b = root_b.getChild("b"); + + assertEquals(1, root_a.getCount()); + assertNull(root_a_a); + assertEquals(1, root_a_b.getCount()); + assertEquals(1, root_b.getCount()); + assertNull(root_b_a); + assertNull(root_b_b); + + assertTrue(root_a_b.isLeaf()); + assertTrue(root_b.isLeaf()); + } + + @Test + public void testEdgeEdge_1() throws Exception { + Edge result = new Edge(); + + assertNotNull(result); + } + + @Test + public void testTrieVertexTrieVertex_1() throws Exception { + String id = "idString"; + + TrieVertex result = new TrieVertex(id); + + assertNotNull(result); + } + + @Test + public void testTrieVertexToString_1() throws Exception { + String id = "idString"; + TrieVertex fixture = new TrieVertex(id); + + String result = fixture.toString(); + + assertEquals(id, result); + } + + @Test + public void testEquals_1() throws Exception { + Trie trieOther = new Trie(); + Trie fixture = new Trie(); + + boolean result = fixture.equals(trieOther); + + assertEquals(true, result); + } + + @Test + public void testEquals_2() throws Exception { + Trie trieOther = new Trie(); + trieOther.train(sequence, 2); + Trie fixture = new Trie(); + fixture.train(sequence, 2); + + boolean result = fixture.equals(trieOther); + + assertEquals(true, result); + } + + @Test + public void testEquals_3() throws Exception { + Trie trieOther = new Trie(); + trieOther.train(sequence, 2); + Trie fixture = new Trie(); + fixture.train(sequence, 3); + + boolean result = fixture.equals(trieOther); + + assertEquals(false, result); + } + + @Test + public void testEquals_4() throws Exception { + Trie trieOther = new Trie(); + Trie fixture = new Trie(); + fixture.train(sequence, 2); + + boolean result = fixture.equals(trieOther); + + assertEquals(false, result); + } + + @Test + public void testEquals_5() throws Exception { + Trie trieOther = new Trie(); + trieOther.train(sequence, 2); + Trie fixture = new Trie(); + + boolean result = fixture.equals(trieOther); + + assertEquals(false, result); + } + + @Test + public void testEquals_6() throws Exception { + Trie fixture = new Trie(); + fixture.train(sequence, 2); + + boolean result = fixture.equals(fixture); + + assertEquals(true, result); + } + + @Test + public void testEquals_7() throws Exception { + Trie fixture = new Trie(); + fixture.train(sequence, 2); + + boolean result = fixture.equals(null); + + assertEquals(false, result); + } + + @Before + public void setUp() throws Exception { + sequence = new ArrayList(); + sequence.add("a"); + sequence.add("b"); + sequence.add("r"); + sequence.add("a"); + sequence.add("c"); + sequence.add("a"); + sequence.add("d"); + sequence.add("a"); + sequence.add("b"); + sequence.add("r"); + sequence.add("a"); + + symbols = new HashSet(); + symbols.add("a"); + symbols.add("b"); + symbols.add("c"); + symbols.add("d"); + symbols.add("r"); + } + + public static void main(String[] args) { + new org.junit.runner.JUnitCore().run(TrieTest.class); + } + }