Test Vicinity.insert method. :param backend_type: The backend type to use. :param vicinity_instance: A Vicinity instance. :param query_vector: A query vector.
(vicinity_instance: Vicinity, query_vector: np.ndarray)
| 81 | |
| 82 | |
| 83 | def test_vicinity_insert(vicinity_instance: Vicinity, query_vector: np.ndarray) -> None: |
| 84 | """ |
| 85 | Test Vicinity.insert method. |
| 86 | |
| 87 | :param backend_type: The backend type to use. |
| 88 | :param vicinity_instance: A Vicinity instance. |
| 89 | :param query_vector: A query vector. |
| 90 | """ |
| 91 | if vicinity_instance.backend.backend_type in {Backend.HNSW, Backend.ANNOY, Backend.PYNNDESCENT}: |
| 92 | # Skip insert for HNSW or Annoy backends. |
| 93 | return |
| 94 | new_item = ["item10001"] |
| 95 | new_vector = query_vector |
| 96 | vicinity_instance.insert(new_item, new_vector[None, :]) |
| 97 | |
| 98 | results = vicinity_instance.query(query_vector, k=50) |
| 99 | |
| 100 | returned_items = [item for item, _ in results[0]] |
| 101 | assert "item10001" in returned_items |
| 102 | |
| 103 | |
| 104 | def test_vicinity_delete(vicinity_instance: Vicinity, items: list[str], vectors: np.ndarray) -> None: |