Function test_vicinity_delete_and_query

tests/test_vicinity.py:286–315

Test Vicinity's delete and query methods together to ensure that indices are correctly handled after deletions. :param vicinity_instance: A Vicinity instance. :param items: List of item names. :param vectors: Array of vectors corresponding to items.

(vicinity_instance: Vicinity, items: list[str], vectors: np.ndarray)

def test_vicinity_delete_and_query(vicinity_instance: Vicinity, items: list[str], vectors: np.ndarray) -> None:
    """
    Test Vicinity's delete and query methods together to ensure that indices are correctly handled after deletions.

    :param vicinity_instance: A Vicinity instance.
    :param items: List of item names.
    :param vectors: Array of vectors corresponding to items.
    """
    if vicinity_instance.backend.backend_type != Backend.BASIC:
        # Skip delete for non-basic backends
        return

    # Delete some items from the Vicinity instance
    non_existing_items_indices = [0, 1, 2]
    items_to_delete = [items[i] for i in non_existing_items_indices]
    vicinity_instance.delete(items_to_delete)

    # Ensure the items are no longer in the items list
    for item in items_to_delete:
        assert item not in vicinity_instance.items

    # Query using a vector of an item that wasn't deleted
    existing_item_index = 3
    item3_vector = vectors[existing_item_index]

    results = vicinity_instance.query(item3_vector, k=10)
    returned_items = [item for item, _ in results[0]]

    # Check that the queried item is in the results
    assert items[existing_item_index] in returned_items


def test_vicinity_evaluate(vicinity_instance: Vicinity, vectors: np.ndarray) -> None: