def test_suffix_empty_trie(self):
     """
     Tests suffix operations on an empty trie.
     """
     trie = naive_suffix_trie.build_suffix_trie("")
     assert naive_suffix_trie.suffix(trie, "")
     assert not naive_suffix_trie.suffix(trie, "a")
 def test_suffix_single_element_trie(self):
     """
     Tests suffix operations on a single element trie.
     """
     trie = naive_suffix_trie.build_suffix_trie("a")
     assert naive_suffix_trie.suffix(trie, "")
     assert naive_suffix_trie.suffix(trie, "a")
     assert not naive_suffix_trie.suffix(trie, "b")
 def test_suffix_multi_element_trie(self):
     """
     Tests suffix operations on a multi-element trie.
     """
     trie = naive_suffix_trie.build_suffix_trie("abba")
     expected = ["", "a", "ba", "bba", "abba"]
     unexpected = ["ab", "abb"]
     for val in expected:
         assert naive_suffix_trie.suffix(trie, val)
     for val in unexpected:
         assert not naive_suffix_trie.suffix(trie, val)