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)