예제 #1
0
파일: test_vtrie.py 프로젝트: uubram/vtrie
def test_keys_items_values():
    t = Trie()
    t[b"foo"] = 5
    assert t.keys() == ["foo"]
    assert t.values() == [5]
    assert t.items() == [("foo", 5)]
    t[b"foobar"] = 3
    t[b"hello"] = 7
    assert set(t.keys()) == set(["foo", "foobar", "hello"])
    assert set(t.values()) == set([5, 3, 7])
    assert set(t.items()) == set([("foo", 5), ("foobar", 3), ("hello", 7)])

    t = Trie()
    n = 1000
    for i in xrange(n):
        t[b(str(i))] = i
    assert len(t.keys()) == n
    values = xrange(n)
    keys = [str(x) for x in values]
    assert set(t.keys()) == set(keys)
    assert set(t.values()) == set(values)
    assert set(t.items()) == set(zip(keys, values))
예제 #2
0
파일: test_vtrie.py 프로젝트: uubram/vtrie
def test_iter():
    """Test iterating over a Trie"""
    t = Trie()
    words = [b"hello", b"foo", b"foobar", b"foozle"]
    for key in words:
        t[key] = 1
    assert set([b(key) for key in t]) == set(words)

    # Modify values in the trie using t's iterator
    for key in t:
        t[b(key)] = 2
    assert t.values() == [2]*len(words)

    # It should be an error to continue to iterate after adding and/or removing
    # nodes.
    i1 = iter(t)
    next(i1) # This should be fine
    t[b"new"] = 1
    with pytest.raises(RuntimeError):
        next(i1)
    # Alright, so addition was detected, now for deletion
    i2 = iter(t)
    del t[b"new"]
    with pytest.raises(RuntimeError):
        next(i2)
    # Detection should not be on number of nodes, so let's add and remove a
    # single node.
    i3 = iter(t)
    next(i3) # should be fine
    n = t.num_nodes()
    t[b"a"] = 1
    assert t.num_nodes() == n + 1
    del t[b"a"]
    assert t.num_nodes() == n
    with pytest.raises(RuntimeError):
        next(i3)