pybbhash/tests/test_bbhash.py at 476b83b65dff0c7da0da1513bc1183f4c287c19a · dib-lab/pybbhash · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
"Test basic bbhash functionality."
import pytest

import bbhash

def test_all(tmpdir):
    x = bbhash.PyMPHF(list(range(10)), 10, 1, 1.0)
    assert x.lookup(9) == 8

    output = tmpdir.join('xxx')
    x.save(str(output))

    y = bbhash.load_mphf(str(output))
    assert y.lookup(9) == 8


def test_lookup():
    x = bbhash.PyMPHF(list(range(10)), 10, 1, 1.0)
    assert all(x.lookup(y) is not None for y in range(10))
    assert x.lookup(200) is None


def test_construct_from_set():
    with pytest.raises(TypeError):        # CTB: could fix this.
        x = bbhash.PyMPHF(set(range(10)), 10, 1, 1.0)


def test_lookup():
    x = bbhash.PyMPHF(list(range(10)), 10, 1, 1.0)

    q = list(range(10))
    y = x.lookup_many(q)

    for i in y:
        assert i is not None


def test_lookup_set():
    x = bbhash.PyMPHF(list(range(10)), 10, 1, 1.0)

    q = set(range(10))
    y = x.lookup_many(q)

    for i in y:
        assert i is not None