word-hunt-solver/trie_test.py at master · taiga-forestry/word-hunt-solver · 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
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
import pytest
import trie

def test_add():
    ''' Tests the add() and contains() methods '''

    # empty trie
    t = trie.Trie()
    assert t.size == 0
    assert t.contains("A", False) == False

    # adding to trie on leaf, empty word case
    t.add("A")
    t.add("A")
    t.add("BOAT")
    assert t.size == 2
    assert t.contains("A", False) == True
    assert t.contains("BOAT", False) == True
    assert t.contains("B", False) == False
    assert t.contains("BOA", False) == False
    assert t.contains("B", True) == True
    assert t.contains("BOA", True) == True

    # adding to trie on node, non-empty word case
    t.add("BOLT")
    assert t.size == 3
    assert t.contains("BOLT", False) == True

    # adding to trie on node, empty word case
    t.add("BOA")
    assert t.size == 4
    assert t.contains("BOA", False) == True

    # adding to trie on leaf, non-empty word case
    t.add("BOATING")
    assert t.size == 5
    assert t.contains("BOATING", False) == True


def test_remove():
    ''' Tests the remove() and contains() methods '''

    # empty trie
    t = trie.Trie()
    assert t.size == 0
    assert t.remove("") == False

    t.add("BOAT")
    t.add("BOLT")
    t.add("BOA")
    t.add("BOATING")

    # removing non-existent element from trie
    assert t.size == 4
    assert t.remove("B") == False

    # removing element twice from trie
    assert t.remove("BOA") == True
    assert t.remove("BOA") == False
    assert t.contains("BOA", False) == False
    assert t.contains("BOA", True) == True
    assert t.size == 3

    # removing element which is "subelement" from trie
    assert t.remove("BOAT") == True
    assert t.contains("BOATING", False) == True
    assert t.contains("BOLT", False) == True
    assert t.contains("BOAT", False) == False
    assert t.contains("BOAT", True) == True
    assert t.size == 2

    # removing element after adding same element
    t.add("BOA")
    assert t.contains("BOA", False) == True
    assert t.contains("BOAT", False) == False
    assert t.contains("BOATING", False) == True
    assert t.contains("BOLT", False) == True
    assert t.size == 3


# function calls!!
test_add()
test_remove()