|
19 | 19 | import org.elasticsearch.common.settings.Settings; |
20 | 20 | import org.elasticsearch.rest.RestStatus; |
21 | 21 | import org.elasticsearch.search.vectors.KnnVectorQueryBuilder; |
22 | | -import org.elasticsearch.search.vectors.VectorData; |
23 | 22 | import org.elasticsearch.test.ESIntegTestCase; |
24 | 23 | import org.elasticsearch.xcontent.XContentBuilder; |
25 | 24 | import org.elasticsearch.xcontent.XContentFactory; |
@@ -131,19 +130,25 @@ public void testDenseVectorMappingUpdate() throws Exception { |
131 | 130 | int expectedDocs = docsBefore + docsAfter; |
132 | 131 |
|
133 | 132 | // Count query |
134 | | - assertNoFailuresAndResponse(client().prepareSearch(INDEX_NAME).setSize(0).setTrackTotalHits(true).setSize(expectedDocs), response -> { |
135 | | - assertHitCount(response, expectedDocs); |
136 | | - }); |
| 133 | + assertNoFailuresAndResponse( |
| 134 | + client().prepareSearch(INDEX_NAME).setSize(0).setTrackTotalHits(true).setSize(expectedDocs), |
| 135 | + response -> { |
| 136 | + assertHitCount(response, expectedDocs); |
| 137 | + } |
| 138 | + ); |
137 | 139 |
|
138 | 140 | // KNN query |
139 | 141 | float[] queryVector = new float[dimensions]; |
140 | 142 | for (int i = 0; i < queryVector.length; i++) { |
141 | 143 | queryVector[i] = randomFloatBetween(-1, 1, true); |
142 | 144 | } |
143 | 145 | KnnVectorQueryBuilder queryBuilder = new KnnVectorQueryBuilder(VECTOR_FIELD, queryVector, null, null, null, null); |
144 | | - assertNoFailuresAndResponse(client().prepareSearch(INDEX_NAME).setQuery(queryBuilder).setTrackTotalHits(true).setSize(expectedDocs), response -> { |
145 | | - assertHitCount(response, expectedDocs); |
146 | | - }); |
| 146 | + assertNoFailuresAndResponse( |
| 147 | + client().prepareSearch(INDEX_NAME).setQuery(queryBuilder).setTrackTotalHits(true).setSize(expectedDocs), |
| 148 | + response -> { |
| 149 | + assertHitCount(response, expectedDocs); |
| 150 | + } |
| 151 | + ); |
147 | 152 | } |
148 | 153 |
|
149 | 154 | private XContentBuilder updateMapping(int dimensions, String type) throws IOException { |
|
0 commit comments