22import unittest
33
44from common import *
5+ from array_paths import *
56
67import tiledb .vector_search as vs
78from tiledb .cloud import groups
@@ -40,8 +41,8 @@ def tearDownClass(cls):
4041
4142 def test_cloud_flat (self ):
4243 source_uri = "tiledb://TileDB-Inc/sift_10k"
43- queries_uri = "test/data/siftsmall/siftsmall_query.fvecs"
44- gt_uri = "test/data/siftsmall/siftsmall_groundtruth.ivecs"
44+ queries_uri = siftsmall_query_file
45+ gt_uri = siftsmall_groundtruth_file
4546 index_uri = CloudTests .flat_index_uri
4647 k = 100
4748 nqueries = 100
@@ -68,8 +69,8 @@ def test_cloud_flat(self):
6869
6970 def test_cloud_ivf_flat (self ):
7071 source_uri = "tiledb://TileDB-Inc/sift_10k"
71- queries_uri = "test/data/siftsmall/siftsmall_query.fvecs"
72- gt_uri = "test/data/siftsmall/siftsmall_groundtruth.ivecs"
72+ queries_uri = siftsmall_query_file
73+ gt_uri = siftsmall_groundtruth_file
7374 index_uri = CloudTests .ivf_flat_index_uri
7475 k = 100
7576 partitions = 100
@@ -162,8 +163,8 @@ def test_cloud_ivf_flat_random_sampling(self):
162163 # source_uri = "tiledb://TileDB-Inc/ann_sift1b_raw_vectors_col_major"
163164 # training_sample_size = 1000000
164165 source_uri = "tiledb://TileDB-Inc/sift_10k"
165- queries_uri = "test/data/siftsmall/siftsmall_query.fvecs"
166- gt_uri = "test/data/siftsmall/siftsmall_groundtruth.ivecs"
166+ queries_uri = siftsmall_query_file
167+ gt_uri = siftsmall_groundtruth_file
167168 index_uri = CloudTests .ivf_flat_random_sampling_index_uri
168169 k = 100
169170 nqueries = 100
0 commit comments