Skip to content

Commit a5a7e34

Browse files
authored
Speed up some slow tests by reducing iteration counts (addresses #15212) (#15214)
1 parent 9ae3844 commit a5a7e34

File tree

4 files changed

+8
-8
lines changed

4 files changed

+8
-8
lines changed

lucene/core/src/test/org/apache/lucene/search/TestFullPrecisionFloatVectorSimilarityValuesSource.java

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -46,8 +46,8 @@ public class TestFullPrecisionFloatVectorSimilarityValuesSource extends LuceneTe
4646
private Codec savedCodec;
4747

4848
private static final String KNN_FIELD = "knnField";
49-
private static final int NUM_VECTORS = 1000;
50-
private static final int VECTOR_DIMENSION = 128;
49+
private static final int NUM_VECTORS = 100;
50+
private static final int VECTOR_DIMENSION = 8;
5151

5252
KnnVectorsFormat format;
5353
Float confidenceInterval;

lucene/core/src/test/org/apache/lucene/search/TestLateInteractionFloatValuesSource.java

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -66,7 +66,7 @@ public void testValidations() {
6666

6767
public void testValues() throws IOException {
6868
List<float[][]> corpus = new ArrayList<>();
69-
final int numDocs = atLeast(1000);
69+
final int numDocs = atLeast(100);
7070
final int numSegments = random().nextInt(2, 5);
7171
final VectorSimilarityFunction vectorSimilarityFunction =
7272
VectorSimilarityFunction.values()[
@@ -96,7 +96,7 @@ public void testValues() throws IOException {
9696
}
9797
}
9898
// add a segment with no vectors
99-
for (int i = 0; i < 100; i++) {
99+
for (int i = 0; i < 10; i++) {
100100
Document doc = new Document();
101101
doc.add(new IntField("has_li_vector", 0, Field.Store.YES));
102102
w.addDocument(doc);

lucene/core/src/test/org/apache/lucene/search/TestLateInteractionRescorer.java

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -130,7 +130,7 @@ public void testMissingLateIValues() throws Exception {
130130
}
131131

132132
private void indexMultiVectors(Directory dir, List<float[][]> corpus) throws IOException {
133-
final int numDocs = atLeast(1000);
133+
final int numDocs = atLeast(100);
134134
final int numSegments = random().nextInt(2, 5);
135135
int id = 0;
136136
try (IndexWriter w = new IndexWriter(dir, newIndexWriterConfig())) {

lucene/core/src/test/org/apache/lucene/search/TestRescoreTopNQuery.java

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -167,9 +167,9 @@ public void testLateInteractionQuery() throws Exception {
167167
final String LATE_I_FIELD = "li_vector";
168168
final String KNN_FIELD = "knn_vector";
169169
List<float[][]> corpus = new ArrayList<>();
170-
final int numDocs = atLeast(1000);
171-
final int numSegments = random().nextInt(2, 10);
172-
final int dim = 128;
170+
final int numDocs = atLeast(100);
171+
final int numSegments = random().nextInt(2, 5);
172+
final int dim = 8;
173173
final VectorSimilarityFunction vectorSimilarityFunction =
174174
VectorSimilarityFunction.values()[
175175
random().nextInt(VectorSimilarityFunction.values().length)];

0 commit comments

Comments
 (0)