Skip to content

Commit

Permalink
Fix spotless
Browse files Browse the repository at this point in the history 
Signed-off-by: Vijayan Balasubramanian <[email protected]>
  • Loading branch information
@VijayanB
VijayanB committed Oct 8, 2024
1 parent 1fe9284 commit 0893e1f
Show file tree
Hide file tree
Showing 2 changed files with 25 additions and 17 deletions.
32 changes: 19 additions & 13 deletions ...rg/opensearch/knn/index/codec/KNN990Codec/NativeEngines990KnnVectorsWriterFlushTests.java
View file
Original file line number Diff line number Diff line change
Expand Up @@ -86,7 +86,7 @@ public class NativeEngines990KnnVectorsWriterFlushTests extends OpenSearchTestCa
public void setUp() throws Exception {
super.setUp();
MockitoAnnotations.openMocks(this);
objectUnderTest = new NativeEngines990KnnVectorsWriter(segmentWriteState, flatVectorsWriter,BUILD_GRAPH_ALWAYS_THRESHOLD);
objectUnderTest = new NativeEngines990KnnVectorsWriter(segmentWriteState, flatVectorsWriter, BUILD_GRAPH_ALWAYS_THRESHOLD);
mockedFlatFieldVectorsWriter = Mockito.mock(FlatFieldVectorsWriter.class);
Mockito.doNothing().when(mockedFlatFieldVectorsWriter).addValue(Mockito.anyInt(), Mockito.any());
Mockito.when(flatVectorsWriter.addField(Mockito.any())).thenReturn(mockedFlatFieldVectorsWriter);
Expand Down Expand Up @@ -343,8 +343,9 @@ public void testFlush_whenThresholdIsNegative_thenNativeIndexWriterIsNeverCalled
);

NativeEngineFieldVectorsWriter field = nativeEngineFieldVectorsWriter(fieldInfo, vectorsPerField.get(i));
fieldWriterMockedStatic.when(() -> NativeEngineFieldVectorsWriter.create(fieldInfo, segmentWriteState.infoStream))
.thenReturn(field);
fieldWriterMockedStatic.when(
() -> NativeEngineFieldVectorsWriter.create(fieldInfo, mockedFlatFieldVectorsWriter, segmentWriteState.infoStream)
).thenReturn(field);
try {
nativeEngineWriter.addField(fieldInfo);
} catch (Exception e) {
Expand Down Expand Up @@ -419,8 +420,9 @@ public void testFlush_whenThresholdIsGreaterThanVectorSize_thenNativeIndexWriter
);

NativeEngineFieldVectorsWriter field = nativeEngineFieldVectorsWriter(fieldInfo, vectorsPerField.get(i));
fieldWriterMockedStatic.when(() -> NativeEngineFieldVectorsWriter.create(fieldInfo, segmentWriteState.infoStream))
.thenReturn(field);
fieldWriterMockedStatic.when(
() -> NativeEngineFieldVectorsWriter.create(fieldInfo, mockedFlatFieldVectorsWriter, segmentWriteState.infoStream)
).thenReturn(field);
try {
nativeEngineWriter.addField(fieldInfo);
} catch (Exception e) {
Expand Down Expand Up @@ -496,8 +498,9 @@ public void testFlush_whenThresholdIsEqualToMinNumberOfVectors_thenNativeIndexWr
);

NativeEngineFieldVectorsWriter field = nativeEngineFieldVectorsWriter(fieldInfo, vectorsPerField.get(i));
fieldWriterMockedStatic.when(() -> NativeEngineFieldVectorsWriter.create(fieldInfo, segmentWriteState.infoStream))
.thenReturn(field);
fieldWriterMockedStatic.when(
() -> NativeEngineFieldVectorsWriter.create(fieldInfo, mockedFlatFieldVectorsWriter, segmentWriteState.infoStream)
).thenReturn(field);
try {
nativeEngineWriter.addField(fieldInfo);
} catch (Exception e) {
Expand Down Expand Up @@ -581,8 +584,9 @@ public void testFlush_whenThresholdIsEqualToFixedValue_thenRelevantNativeIndexWr
);

NativeEngineFieldVectorsWriter field = nativeEngineFieldVectorsWriter(fieldInfo, vectorsPerField.get(i));
fieldWriterMockedStatic.when(() -> NativeEngineFieldVectorsWriter.create(fieldInfo, segmentWriteState.infoStream))
.thenReturn(field);
fieldWriterMockedStatic.when(
() -> NativeEngineFieldVectorsWriter.create(fieldInfo, mockedFlatFieldVectorsWriter, segmentWriteState.infoStream)
).thenReturn(field);
try {
nativeEngineWriter.addField(fieldInfo);
} catch (Exception e) {
Expand Down Expand Up @@ -669,8 +673,9 @@ public void testFlush_whenQuantizationIsProvided_whenBuildGraphDatStructureThres
);

NativeEngineFieldVectorsWriter field = nativeEngineFieldVectorsWriter(fieldInfo, vectorsPerField.get(i));
fieldWriterMockedStatic.when(() -> NativeEngineFieldVectorsWriter.create(fieldInfo, segmentWriteState.infoStream))
.thenReturn(field);
fieldWriterMockedStatic.when(
() -> NativeEngineFieldVectorsWriter.create(fieldInfo, mockedFlatFieldVectorsWriter, segmentWriteState.infoStream)
).thenReturn(field);

try {
nativeEngineWriter.addField(fieldInfo);
Expand Down Expand Up @@ -771,8 +776,9 @@ public void testFlush_whenQuantizationIsProvided_whenBuildGraphDatStructureThres
);

NativeEngineFieldVectorsWriter field = nativeEngineFieldVectorsWriter(fieldInfo, vectorsPerField.get(i));
fieldWriterMockedStatic.when(() -> NativeEngineFieldVectorsWriter.create(fieldInfo, segmentWriteState.infoStream))
.thenReturn(field);
fieldWriterMockedStatic.when(
() -> NativeEngineFieldVectorsWriter.create(fieldInfo, mockedFlatFieldVectorsWriter, segmentWriteState.infoStream)
).thenReturn(field);

try {
nativeEngineWriter.addField(fieldInfo);
Expand Down
10 changes: 6 additions & 4 deletions ...rg/opensearch/knn/index/codec/KNN990Codec/NativeEngines990KnnVectorsWriterMergeTests.java
View file
Original file line number Diff line number Diff line change
Expand Up @@ -196,8 +196,9 @@ public void testMerge_whenThresholdIsNegative_thenNativeIndexWriterIsNeverCalled
);

NativeEngineFieldVectorsWriter field = nativeEngineFieldVectorsWriter(fieldInfo, mergedVectors);
fieldWriterMockedStatic.when(() -> NativeEngineFieldVectorsWriter.create(fieldInfo, segmentWriteState.infoStream))
.thenReturn(field);
fieldWriterMockedStatic.when(
() -> NativeEngineFieldVectorsWriter.create(fieldInfo, mockedFlatFieldVectorsWriter, segmentWriteState.infoStream)
).thenReturn(field);

mergedVectorValuesMockedStatic.when(() -> KnnVectorsWriter.MergedVectorValues.mergeFloatVectorValues(fieldInfo, mergeState))
.thenReturn(floatVectorValues);
Expand Down Expand Up @@ -253,8 +254,9 @@ public void testMerge_whenThresholdIsEqualToNumberOfVectors_thenNativeIndexWrite
);

NativeEngineFieldVectorsWriter field = nativeEngineFieldVectorsWriter(fieldInfo, mergedVectors);
fieldWriterMockedStatic.when(() -> NativeEngineFieldVectorsWriter.create(fieldInfo, segmentWriteState.infoStream))
.thenReturn(field);
fieldWriterMockedStatic.when(
() -> NativeEngineFieldVectorsWriter.create(fieldInfo, mockedFlatFieldVectorsWriter, segmentWriteState.infoStream)
).thenReturn(field);

mergedVectorValuesMockedStatic.when(() -> KnnVectorsWriter.MergedVectorValues.mergeFloatVectorValues(fieldInfo, mergeState))
.thenReturn(floatVectorValues);
Expand Down

0 comments on commit 0893e1f

Please sign in to comment.