Java/src/main/java/com/thealgorithms/datastructures/trees/TernarySearchTreeTest.java at 237daaa41ad56b7f2cae9832fc1751284cb4944a · Sarabjeet13/Java · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
package main.java.com.thealgorithms.datastructures.trees;

import org.junit.jupiter.api.Test;
import static org.junit.jupiter.api.Assertions.*;

/**
 * Unit tests for TernarySearchTree.
 */
class TernarySearchTreeTest {

    @Test
    void testInsertAndSearchWords() {
        TernarySearchTree tst = new TernarySearchTree();

        tst.insert("cat");
        tst.insert("cap");
        tst.insert("bat");

        assertTrue(tst.search("cat"));
        assertTrue(tst.search("cap"));
        assertTrue(tst.search("bat"));

        assertFalse(tst.search("can"));
        assertFalse(tst.search("dog"));
    }

    @Test
    void testSingleCharacterWord() {
        TernarySearchTree tst = new TernarySearchTree();

        tst.insert("a");
        tst.insert("b");

        assertTrue(tst.search("a"));
        assertTrue(tst.search("b"));
        assertFalse(tst.search("c"));
    }

    @Test
    void testPrefixNotWholeWord() {
        TernarySearchTree tst = new TernarySearchTree();

        tst.insert("hello");

        assertFalse(tst.search("he"));     // prefix only
        assertFalse(tst.search("hell"));   // prefix only
        assertTrue(tst.search("hello"));   // complete word
    }

    @Test
    void testMultipleInsertionsOfSameWord() {
        TernarySearchTree tst = new TernarySearchTree();

        tst.insert("java");
        tst.insert("java");
        tst.insert("java");

        assertTrue(tst.search("java"));
        assertFalse(tst.search("javac"));
    }
}