annoy
1# Copyright (c) 2013 Spotify AB
2#
3# Licensed under the Apache License, Version 2.0 (the "License"); you may not
4# use this file except in compliance with the License. You may obtain a copy of
5# the License at
6#
7# http://www.apache.org/licenses/LICENSE-2.0
8#
9# Unless required by applicable law or agreed to in writing, software
10# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
11# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
12# License for the specific language governing permissions and limitations under
13# the License.
14
15import multiprocessing.pool
16
17import numpy
18
19from annoy import AnnoyIndex
20
21
22def test_threads():
23n, f = 10000, 10
24i = AnnoyIndex(f, "euclidean")
25for j in range(n):
26i.add_item(j, numpy.random.normal(size=f))
27i.build(10)
28
29pool = multiprocessing.pool.ThreadPool()
30
31def query_f(j):
32i.get_nns_by_item(1, 1000)
33
34pool.map(query_f, range(n))
35