2020
2121from vector_search import store_vectors
2222from vector_search import vector_search_basic
23- from vector_search import vector_search_distance_result_field
23+ from vector_search import vector_search_distance_result_property
2424from vector_search import vector_search_distance_threshold
2525from vector_search import vector_search_prefilter
26+ from vector_search import vector_search_large_response
2627
2728
2829os .environ ["GOOGLE_CLOUD_PROJECT" ] = os .environ ["FIRESTORE_PROJECT" ]
@@ -85,8 +86,8 @@ def test_vector_search_prefilter(db):
8586 assert results [1 ].key .name == "Excelsa"
8687
8788
88- def test_vector_search_distance_result_field (db ):
89- vector_query = vector_search_distance_result_field (db )
89+ def test_vector_search_distance_result_property (db ):
90+ vector_query = vector_search_distance_result_property (db )
9091 results = list (vector_query .fetch ())
9192
9293 assert len (results ) == 4
@@ -108,8 +109,8 @@ def test_vector_search_distance_threshold(db):
108109 assert results [0 ].key .name == "Liberica"
109110 assert results [1 ].key .name == "Robusta"
110111
111- def test_vector_search_large_query (db ):
112- key_list , vector_results , full_results = vector_search_large_query (db )
112+ def test_vector_search_large_response (db ):
113+ key_list , vector_results , full_results = vector_search_large_response (db )
113114 assert len (key_list ) == 4
114115 # each list should have same number of elements
115116 assert len (key_list ) == len (vector_results )
0 commit comments