@Test public void testCopyPositions() { Slice[] expectedValues = alternatingNullValues(createTestValue(17)); BlockBuilder blockBuilder = createBlockBuilderWithValues(expectedValues); assertBlockFilteredPositions(expectedValues, blockBuilder.build(), () -> blockBuilder.newBlockBuilderLike(null), 0, 2, 4, 6, 7, 9, 10, 16); }
@Test public void testEstimatedDataSizeForStats() { Slice[] expectedValues = createTestValue(100); assertEstimatedDataSizeForStats(createBlockBuilderWithValues(expectedValues), expectedValues); }
@Test public void test() { Slice[] expectedValues = createTestValue(17); assertFixedWithValues(expectedValues); assertFixedWithValues(alternatingNullValues(expectedValues)); }
private void assertFixedWithValues(Slice[] expectedValues) { BlockBuilder blockBuilder = createBlockBuilderWithValues(expectedValues); assertBlock(blockBuilder, () -> blockBuilder.newBlockBuilderLike(null), expectedValues); assertBlock(blockBuilder.build(), () -> blockBuilder.newBlockBuilderLike(null), expectedValues); }
@Test public void testCompactBlock() { short[] shortArray = {(short) 0, (short) 0, (short) 1, (short) 2, (short) 3, (short) 4}; boolean[] valueIsNull = {false, true, false, false, false, false}; testCompactBlock(new ShortArrayBlock(0, Optional.empty(), new short[0])); testCompactBlock(new ShortArrayBlock(shortArray.length, Optional.of(valueIsNull), shortArray)); testIncompactBlock(new ShortArrayBlock(shortArray.length - 1, Optional.of(valueIsNull), shortArray)); }
@Test public void testLazyBlockBuilderInitialization() { Slice[] expectedValues = createTestValue(100); BlockBuilder emptyBlockBuilder = new ShortArrayBlockBuilder(null, 0); BlockBuilder blockBuilder = new ShortArrayBlockBuilder(null, expectedValues.length); assertEquals(blockBuilder.getSizeInBytes(), emptyBlockBuilder.getSizeInBytes()); assertEquals(blockBuilder.getRetainedSizeInBytes(), emptyBlockBuilder.getRetainedSizeInBytes()); writeValues(expectedValues, blockBuilder); assertTrue(blockBuilder.getSizeInBytes() > emptyBlockBuilder.getSizeInBytes()); assertTrue(blockBuilder.getRetainedSizeInBytes() > emptyBlockBuilder.getRetainedSizeInBytes()); blockBuilder = blockBuilder.newBlockBuilderLike(null); assertEquals(blockBuilder.getSizeInBytes(), emptyBlockBuilder.getSizeInBytes()); assertEquals(blockBuilder.getRetainedSizeInBytes(), emptyBlockBuilder.getRetainedSizeInBytes()); }
private static Slice[] createTestValue(int positionCount) { Slice[] expectedValues = new Slice[positionCount]; for (int position = 0; position < positionCount; position++) { expectedValues[position] = createExpectedValue(SIZE_OF_SHORT); } return expectedValues; }