Skip to content

Commit 4653234

Browse files
authored
Update URIs in test_cloud.py to match other unit tests (#277)
1 parent b7c6521 commit 4653234

File tree

1 file changed

+7
-6
lines changed

1 file changed

+7
-6
lines changed

apis/python/test/test_cloud.py

Lines changed: 7 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -2,6 +2,7 @@
22
import unittest
33

44
from common import *
5+
from array_paths import *
56

67
import tiledb.vector_search as vs
78
from 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

Comments
 (0)