Function test_vicinity_insert_mismatched_lengths

tests/test_vicinity.py:258–269

Test that Vicinity.insert raises ValueError when tokens and vectors lengths do not match. :param vicinity_instance: A Vicinity instance. :raises ValueError: If tokens and vectors lengths differ.

(vicinity_instance: Vicinity, query_vector: np.ndarray)

def test_vicinity_insert_mismatched_lengths(vicinity_instance: Vicinity, query_vector: np.ndarray) -> None:
    """
    Test that Vicinity.insert raises ValueError when tokens and vectors lengths do not match.

    :param vicinity_instance: A Vicinity instance.
    :raises ValueError: If tokens and vectors lengths differ.
    """
    new_items = ["item10002", "item10003"]
    new_vector = query_vector

    with pytest.raises(ValueError):
        vicinity_instance.insert(new_items, new_vector[None, :])


def test_vicinity_insert_wrong_dimension(vicinity_instance: Vicinity) -> None: