Skip to content

Commit 343e906

Browse files
Update test script to test query cycling optimization
1 parent 6127e84 commit 343e906

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

test_multiprocessing.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -20,15 +20,15 @@ def _search_one(cls, query, top=None):
2020
def setup_search(self):
2121
pass
2222

23-
# Create test queries
24-
queries = [Query(vector=[0.1]*10, meta_conditions=None, expected_result=None) for _ in range(1000)]
23+
# Create a small set of test queries
24+
queries = [Query(vector=[0.1]*10, meta_conditions=None, expected_result=None) for _ in range(10)]
2525

2626
# Create a searcher with parallel=10
2727
searcher = TestSearcher('localhost', {}, {'parallel': 10})
2828

29-
# Run the search_all method
29+
# Run the search_all method with a large num_queries parameter
3030
start = time.perf_counter()
31-
results = searcher.search_all('cosine', queries)
31+
results = searcher.search_all('cosine', queries, num_queries=1000)
3232
total_time = time.perf_counter() - start
3333

3434
print(f'Number of queries: {len(results["latencies"])}')

0 commit comments

Comments
 (0)