hannoy 0.3.0

HNSW Approximate Nearest Neighbors in Rust, based on LMDB and optimized for memory usage
Documentation 
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
46
47
48
49
50
51
52
53
54
55
56
57

from pathlib import Path

import numpy as np
import pytest

import hannoy
from hannoy import Reader

def test_by_items_matches_by_item(array_db: hannoy.Database) -> None:
    reader: Reader = array_db.reader(0)
    batch = reader.by_items([1, 2, 3], n = 2, ef_search = 10)
    per_item = [reader.by_item(i, n = 2, ef_search= 10) for i in (1, 2, 3)]
    assert batch == per_item

def test_by_items_batches_query(array_db: hannoy.Database) -> None:
    reader: Reader = array_db.reader(0)
    res = reader.by_items([0, 1, 2], n = 2, ef_search = 10)
    assert len(res) == 3
    assert [sorted(i for i, _ in row) for row in res] == [[1, 2], [0, 2], [0, 1]]



def test_by_items_batches_queries(array_db: hannoy.Database) -> None:
    reader: Reader = array_db.reader(0)
    res = reader.by_items([0, 1], n=2, ef_search=10)
    assert len(res) == 2
    assert [sorted(i for i, _ in row) for row in res] == [[1, 2], [0, 2]]


def test_by_items_none_for_missing_id(array_db: hannoy.Database) -> None:
    reader: Reader = array_db.reader(0)
    res = reader.by_items([0, 999], n=2, ef_search=10)

    assert res[0] is not None
    assert res[1] is None


def test_by_items_preserves_order_with_missing(array_db: hannoy.Database) -> None:
    reader: Reader = array_db.reader(0)
    res = reader.by_items([0, 999, 1], n=2, ef_search=10)
    assert len(res) == 3
    assert res[0] is not None
    assert res[1] is None
    assert res[2] is not None


def test_by_items_duplicate_ids(array_db: hannoy.Database) -> None:
    reader: Reader = array_db.reader(0)
    res = reader.by_items([0, 0, 1], n=2, ef_search=10)
    assert len(res) == 3
    assert res[0] == res[1]
    assert res[0] == reader.by_item(0, n=2, ef_search=10)


def test_by_items_empty(array_db: hannoy.Database) -> None:
    reader: Reader = array_db.reader(0)
    assert reader.by_items([], n=2, ef_search=10) == []