@Test
public void testSimple() {
Trie trie = instance();
assertTrue(trie.add("abc"));
assertTrue(trie.contains("abc"));
assertEquals(1, trie.size());
assertEquals(1, trie.howManyStartsWithPrefix("abc"));
}
@Test(expected = IOException.class)
public void testSimpleSerializationFails() throws IOException {
Trie trie = instance();
assertTrue(trie.add("abc"));
assertTrue(trie.add("cde"));
OutputStream outputStream =
new OutputStream() {
@Override
public void write(int b) throws IOException {
throw new IOException("Fail");
}
};
((StreamSerializable) trie).serialize(outputStream);
}
@org.junit.Test
public void testAdd() throws Exception {
String str1 = "cat";
String str2 = "dog";
Trie trie = new TrieImpl();
assertTrue(trie.add(str1));
assertEquals(1, trie.size());
assertTrue(trie.add(str2));
assertFalse(trie.add(str1));
assertFalse(trie.add(str2));
assertTrue(trie.add("ca"));
assertTrue(trie.add("c"));
assertTrue(trie.add(""));
}
@Test
public void testSimpleSerialization() throws IOException {
Trie trie = instance();
assertTrue(trie.add("abc"));
assertTrue(trie.add("cde"));
assertEquals(2, trie.size());
ByteArrayOutputStream outputStream = new ByteArrayOutputStream();
((StreamSerializable) trie).serialize(outputStream);
ByteArrayInputStream inputStream = new ByteArrayInputStream(outputStream.toByteArray());
Trie newTrie = instance();
((StreamSerializable) newTrie).deserialize(inputStream);
assertTrue(newTrie.contains("abc"));
assertTrue(newTrie.contains("cde"));
assertEquals(2, trie.size());
}
@org.junit.Test
public void testRemove() throws Exception {
Trie trie = new TrieImpl();
trie.add("String");
trie.add("Word");
assertEquals(2, trie.size());
assertFalse(trie.remove("YetWord"));
assertEquals(2, trie.size());
assertTrue(trie.remove("Word"));
assertEquals(1, trie.size());
assertFalse(trie.contains("Word"));
trie.add("Word");
trie.add("word");
assertTrue(trie.remove("word"));
assertEquals(2, trie.size());
assertFalse(trie.contains("word"));
assertFalse(trie.contains(""));
assertFalse(trie.remove(""));
assertEquals(2, trie.size());
trie.add("");
assertEquals(3, trie.size());
assertTrue(trie.remove(""));
assertEquals(2, trie.size());
assertFalse(trie.contains(""));
trie.remove("Word");
trie.remove("String");
assertEquals(0, trie.size());
trie.add("Very");
trie.add("VeryVery");
trie.add("VeryVeryLong");
trie.add("VeryVeryLongWord");
assertEquals(4, trie.size());
trie.remove("Very");
assertEquals(3, trie.size());
assertFalse(trie.contains("Very"));
assertTrue(trie.contains("VeryVery"));
trie.remove("VeryVery");
assertEquals(2, trie.size());
assertFalse(trie.contains("VeryVery"));
assertTrue(trie.contains("VeryVeryLong"));
trie.add("VeryVeryLongWord1");
assertEquals(3, trie.size());
trie.remove("VeryVeryLongWord");
assertEquals(2, trie.size());
assertTrue(trie.contains("VeryVeryLongWord1"));
}
@org.junit.Test
public void testContains() throws Exception {
Trie trie = new TrieImpl();
trie.add("Very");
trie.add("VeryVery");
trie.add("VeryVeryLong");
trie.add("VeryVeryLongWord");
assertTrue(trie.contains("Very"));
assertTrue(trie.contains("VeryVery"));
assertTrue(trie.contains("VeryVeryLong"));
assertTrue(trie.contains("VeryVeryLongWord"));
assertFalse(trie.contains("AnyWord"));
assertFalse(trie.contains("Word"));
assertFalse(trie.contains("Ve"));
assertFalse(trie.contains("VeryV"));
assertFalse(trie.contains("VeryVeryLongWord1"));
assertFalse(trie.contains("VeryVeryLongWor"));
assertFalse(trie.contains(""));
trie.add("");
assertTrue(trie.contains(""));
}
@org.junit.Test
public void testHowManyStartsWithPrefix() throws Exception {
Trie trie = new TrieImpl();
trie.add("Very");
trie.add("VeryVery");
trie.add("VeryVeryLong");
trie.add("VeryVeryLongWord");
assertEquals(4, trie.howManyStartsWithPrefix("V"));
assertEquals(4, trie.howManyStartsWithPrefix("Very"));
assertEquals(0, trie.howManyStartsWithPrefix("VE"));
assertEquals(3, trie.howManyStartsWithPrefix("VeryV"));
assertEquals(3, trie.howManyStartsWithPrefix("VeryVery"));
assertEquals(0, trie.howManyStartsWithPrefix("VeryVeryV"));
assertEquals(2, trie.howManyStartsWithPrefix("VeryVeryL"));
assertEquals(1, trie.howManyStartsWithPrefix("VeryVeryLongWord"));
assertEquals(4, trie.howManyStartsWithPrefix(""));
}
@org.junit.Test
public void testSize() throws Exception {
Trie trie = new TrieImpl();
assertEquals(0, trie.size());
trie.add("String");
assertEquals(1, trie.size());
trie.add("String2");
assertEquals(2, trie.size());
trie.add("Word");
assertEquals(3, trie.size());
trie.remove("str");
assertEquals(3, trie.size());
trie.remove("String");
assertEquals(2, trie.size());
trie.remove("String2");
assertEquals(1, trie.size());
trie.remove("Word");
assertEquals(0, trie.size());
trie.remove("Word");
assertEquals(0, trie.size());
trie.add("");
assertEquals(1, trie.size());
trie.add("");
assertEquals(1, trie.size());
trie.remove("");
assertEquals(0, trie.size());
}