def test_insert(): trie = ZTrie() trie.insert("中国人", counter=3, end=True) trie.insert("中国节", counter=100, end=False) assert trie.size() == 5 assert trie.get("中") == (False, 1) assert trie.get("中国") == (False, 1) assert trie.get("中国人") == (True, 3) assert trie.get("中国节") == (False, 100) assert trie.get("呵呵") is None
def test_equal(): t1 = ZTrie() t1.insert("中国人", counter=3, end=True) t1.insert("中国节", counter=100, end=False) t2 = ZTrie() t2.insert("中国人", counter=3, end=True) t2.insert("中国节", counter=100, end=False) assert t1.equal(t1) assert t1.equal(t2) assert t2.equal(t2) assert t2.equal(t1) t2.add("中国人") assert not t1.equal(t2) assert t1.equal(t2, counter=False, end=False) t2.add("中国菜") assert not t1.equal(t2)