Test Vicinity.save and Vicinity.load. :param tmp_path: Temporary directory provided by pytest. :param vicinity_instance: A Vicinity instance.
(tmp_path: Path, vicinity_instance: Vicinity)
| 132 | |
| 133 | |
| 134 | def test_vicinity_save_and_load(tmp_path: Path, vicinity_instance: Vicinity) -> None: |
| 135 | """ |
| 136 | Test Vicinity.save and Vicinity.load. |
| 137 | |
| 138 | :param tmp_path: Temporary directory provided by pytest. |
| 139 | :param vicinity_instance: A Vicinity instance. |
| 140 | """ |
| 141 | save_path = tmp_path / "vicinity_data" |
| 142 | vicinity_instance.save(save_path) |
| 143 | assert vicinity_instance.vector_store is None |
| 144 | |
| 145 | v = Vicinity.load(save_path) |
| 146 | assert v.vector_store is None |
| 147 | |
| 148 | |
| 149 | def test_vicinity_save_and_load_vector_store(tmp_path: Path, vicinity_instance_with_stored_vectors: Vicinity) -> None: |