def testTrieVocabulary(self): word = "alpha" t = TrieVocabulary() t.addWord(word) t.addWord("spam") t.addWord("ham") t.addWord("eggs") assert t.isPrefix(word[0:2]) assert not t.isPrefix(word) assert t.isWord(word) assert not t.isWord(word[0:2]) assert not t.isPrefix(word[1:]) pass
def testTrieVocabularyFromFile(self): vFile = open('../resources/sowpods.txt','r') vocabulary = TrieVocabulary() for _ in range(50): vocabulary.addWord(vFile.readline()[:-1]) vFile.close() vFile = open('../resources/sowpods.txt','r') for _ in range(50): assert vocabulary.isWord(vFile.readline()[:-1]) vFile.close()