@@ -207,18 +207,12 @@ public void testParallelVectorLoading() throws IOException {
207207
208208                // Get initial documents 
209209                TopDocs  topDocs  = searcher .search (new  MatchAllDocsQuery (), 20 );
210-                 int [] docIds  = Arrays .stream (topDocs .scoreDocs )
211-                     .mapToInt (scoreDoc  -> scoreDoc .doc )
212-                     .toArray ();
210+                 int [] docIds  = Arrays .stream (topDocs .scoreDocs ).mapToInt (scoreDoc  -> scoreDoc .doc ).toArray ();
213211
214212                // Test parallel loading 
215213                DirectIOVectorBatchLoader  batchLoader  = new  DirectIOVectorBatchLoader ();
216214
217-                 Map <Integer , float []> parallelResult  = batchLoader .loadSegmentVectors (
218-                     docIds ,
219-                     reader .leaves ().get (0 ),
220-                     VECTOR_FIELD 
221-                 );
215+                 Map <Integer , float []> parallelResult  = batchLoader .loadSegmentVectors (docIds , reader .leaves ().get (0 ), VECTOR_FIELD );
222216
223217                // use regular vector loader 
224218                Map <Integer , float []> sequentialResult  = new  HashMap <>();
@@ -229,16 +223,17 @@ public void testParallelVectorLoading() throws IOException {
229223                // Verify results are identical 
230224                assertThat (
231225                    "Parallel and sequential results should have same size" ,
232-                     parallelResult .size (), equalTo (sequentialResult .size ()));
226+                     parallelResult .size (),
227+                     equalTo (sequentialResult .size ())
228+                 );
233229
234230                for  (int  docId  : docIds ) {
235231                    float [] parallelVector  = parallelResult .get (docId );
236232                    float [] sequentialVector  = sequentialResult .get (docId );
237233
238234                    assertNotNull ("Parallel result should contain vector for doc "  + docId , parallelVector );
239235                    assertNotNull ("Sequential result should contain vector for doc "  + docId , sequentialVector );
240-                     assertArrayEquals ("Vectors should be identical for doc "  + docId ,
241-                         sequentialVector , parallelVector , 0.0001f );
236+                     assertArrayEquals ("Vectors should be identical for doc "  + docId , sequentialVector , parallelVector , 0.0001f );
242237                }
243238            }
244239        }
0 commit comments