Function test_vicinity_evaluate

tests/test_vicinity.py:318–335

Test the evaluate method of the Vicinity instance. :param vicinity_instance: A Vicinity instance. :param vectors: The full dataset vectors used to build the index.

(vicinity_instance: Vicinity, vectors: np.ndarray)

def test_vicinity_evaluate(vicinity_instance: Vicinity, vectors: np.ndarray) -> None:
    """
    Test the evaluate method of the Vicinity instance.

    :param vicinity_instance: A Vicinity instance.
    :param vectors: The full dataset vectors used to build the index.
    """
    query_vectors = vectors[:10]
    qps, recall = vicinity_instance.evaluate(vectors, query_vectors)

    # Ensure the QPS and recall values are within valid ranges
    assert qps > 0
    assert 0 <= recall <= 1

    # Test with an unsupported metric
    vicinity_instance.backend.arguments.metric = "manhattan"
    with pytest.raises(ValueError):
        vicinity_instance.evaluate(vectors, query_vectors)