annoy

1
import numpy
2

3
from annoy import AnnoyIndex
4

5

6
def _test_building_with_threads(n_jobs):
7
    n, f = 10000, 10
8
    n_trees = 31
9
    i = AnnoyIndex(f, "euclidean")
10
    for j in range(n):
11
        i.add_item(j, numpy.random.normal(size=f))
12
    assert i.build(n_trees, n_jobs=n_jobs)
13
    assert n_trees == i.get_n_trees()
14

15

16
def test_one_thread():
17
    _test_building_with_threads(1)
18

19

20
def test_two_threads():
21
    _test_building_with_threads(2)
22

23

24
def test_four_threads():
25
    _test_building_with_threads(4)
26

27

28
def test_eight_threads():
29
    _test_building_with_threads(8)
30