michael-rapp / tries Goto Github PK

import de.mrapp.tries.PatriciaTrie;
import de.mrapp.tries.sequence.StringSequence;
import org.junit.Test;

import java.util.*;

import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertNull;

public class TrieTest {

  @Test
  public void test1() {
    PatriciaTrie<StringSequence, String> trie = new PatriciaTrie<>();

    trie.put(s("foo"), "foo");
    assertEquals("foo", trie.get(s("foo")));
    assertNull(trie.get(null));

    trie.put(s(""), "empty");
    assertEquals("empty", trie.get(s("")));
    assertNull(trie.get(null)); // fail
    assertEquals(new HashSet<>(Arrays.asList(s("foo"), s(""))), trie.keySet()); // fail

    trie.put(null, "null");
    assertEquals("null", trie.get(null));
    assertEquals("empty", trie.get(s(""))); // fail
    assertEquals(new HashSet<>(Arrays.asList(s("foo"), s(""), null)), trie.keySet()); // fail
  }

  private static StringSequence s(String str) {
    return new StringSequence(str);
  }

}