def test_prefix(self): obj = DTrie() actual_words = ['abx','abc','abcd','bbc'] [obj.add(w) for w in actual_words] for w in actual_words: self.assertTrue(obj.isWord(w)) self.assertEqual(len(obj.getAllWords()),4) for w in actual_words: self.assertFalse( obj.isWord( w+'NA' ) ) self.assertFalse( obj.isWord( w+'DA' ) ) for pfx in ["ab","bb","bd"]: self.assertFalse( obj.isWord( pfx ) ) for pfx in ['a','ab','b','bb']: self.assertTrue( obj.hasWordPrefix(pfx) ) val = [] for pfx in ['c','ac','ad','cb','z']: #print("===> last/test ==> %s"%pfx) val.append( obj.hasWordPrefix(pfx) ) self.assertFalse( any(val) ) return
def test_prefix(self): obj = DTrie() actual_words = ["abx", "abc", "abcd", "bbc"] [obj.add(w) for w in actual_words] for w in actual_words: self.assertTrue(obj.isWord(w)) self.assertEqual(len(obj.getAllWords()), 4) for w in actual_words: self.assertFalse(obj.isWord(w + "NA")) self.assertFalse(obj.isWord(w + "DA")) for pfx in ["ab", "bb", "bd"]: self.assertFalse(obj.isWord(pfx)) for pfx in ["a", "ab", "b", "bb"]: self.assertTrue(obj.hasWordPrefix(pfx)) val = [] for pfx in ["c", "ac", "ad", "cb", "z"]: # print("===> last/test ==> %s"%pfx) val.append(obj.hasWordPrefix(pfx)) self.assertFalse(any(val)) return