@Test
public void testStackRepresentation() throws Exception {
Block array = arrayBlockOf(BIGINT, 1L, 2L);
Block actualBlock = mapBlockOf(DOUBLE, new ArrayType(BIGINT), ImmutableMap.of(1.0, array));
DynamicSliceOutput actualSliceOutput = new DynamicSliceOutput(100);
writeBlock(actualSliceOutput, actualBlock);
Block expectedBlock =
new InterleavedBlockBuilder(
ImmutableList.<Type>of(DOUBLE, new ArrayType(BIGINT)), new BlockBuilderStatus(), 3)
.writeDouble(1.0)
.closeEntry()
.writeObject(
BIGINT
.createBlockBuilder(new BlockBuilderStatus(), 1)
.writeLong(1L)
.closeEntry()
.writeLong(2L)
.closeEntry()
.build())
.closeEntry()
.build();
DynamicSliceOutput expectedSliceOutput = new DynamicSliceOutput(100);
writeBlock(expectedSliceOutput, expectedBlock);
assertEquals(actualSliceOutput.slice(), expectedSliceOutput.slice());
}
@Override
public Block getSequenceBlock(int start, int length) {
BlockBuilder blockBuilder = BIGINT.createBlockBuilder(new BlockBuilderStatus());
for (int i = start; i < start + length; i++) {
BIGINT.writeLong(blockBuilder, i);
}
return blockBuilder.build();
}
private static Block getRowNumberColumn(List<Page> pages) {
BlockBuilder builder = BIGINT.createBlockBuilder(new BlockBuilderStatus());
for (Page page : pages) {
int rowNumberChannel = page.getChannelCount() - 1;
for (int i = 0; i < page.getPositionCount(); i++) {
BIGINT.writeLong(builder, page.getLong(BIGINT, rowNumberChannel, i));
}
}
return builder.build();
}
public class TestApproximatePercentileAggregation {
private static final Block EMPTY_DOUBLE_BLOCK =
DOUBLE.createBlockBuilder(new BlockBuilderStatus(), 0).build();
private static final Block EMPTY_LONG_BLOCK =
BIGINT.createBlockBuilder(new BlockBuilderStatus(), 0).build();
@Test
public void testLongPartialStep() throws Exception {
// regular approx_percentile
assertAggregation(
LONG_APPROXIMATE_PERCENTILE_AGGREGATION,
1.0,
null,
createLongsBlock(null, null),
createRLEBlock(0.5, 2));
assertAggregation(
LONG_APPROXIMATE_PERCENTILE_AGGREGATION,
1.0,
1L,
createLongsBlock(null, 1L),
createRLEBlock(0.5, 2));
assertAggregation(
LONG_APPROXIMATE_PERCENTILE_AGGREGATION,
1.0,
2L,
createLongsBlock(null, 1L, 2L, 3L),
createRLEBlock(0.5, 4));
assertAggregation(
LONG_APPROXIMATE_PERCENTILE_AGGREGATION,
1.0,
2L,
createLongsBlock(1L, 2L, 3L),
createRLEBlock(0.5, 3));
assertAggregation(
LONG_APPROXIMATE_PERCENTILE_AGGREGATION,
1.0,
3L,
createLongsBlock(
1L, null, 2L, 2L, null, 2L, 2L, null, 2L, 2L, null, 3L, 3L, null, 3L, null, 3L, 4L, 5L,
6L, 7L),
createRLEBlock(0.5, 21));
// weighted approx_percentile
assertAggregation(
LONG_APPROXIMATE_PERCENTILE_WEIGHTED_AGGREGATION,
1.0,
null,
createLongsBlock(null, null),
createLongsBlock(1L, 1L),
createRLEBlock(0.5, 2));
assertAggregation(
LONG_APPROXIMATE_PERCENTILE_WEIGHTED_AGGREGATION,
1.0,
1L,
createLongsBlock(null, 1L),
createLongsBlock(1L, 1L),
createRLEBlock(0.5, 2));
assertAggregation(
LONG_APPROXIMATE_PERCENTILE_WEIGHTED_AGGREGATION,
1.0,
2L,
createLongsBlock(null, 1L, 2L, 3L),
createLongsBlock(1L, 1L, 1L, 1L),
createRLEBlock(0.5, 4));
assertAggregation(
LONG_APPROXIMATE_PERCENTILE_WEIGHTED_AGGREGATION,
1.0,
2L,
createLongsBlock(1L, 2L, 3L),
createLongsBlock(1L, 1L, 1L),
createRLEBlock(0.5, 3));
assertAggregation(
LONG_APPROXIMATE_PERCENTILE_WEIGHTED_AGGREGATION,
1.0,
3L,
createLongsBlock(
1L, null, 2L, null, 2L, null, 2L, null, 3L, null, 3L, null, 3L, 4L, 5L, 6L, 7L),
createLongsBlock(1L, 1L, 2L, 1L, 2L, 1L, 2L, 1L, 2L, 1L, 1L, 1L, 1L, 1L, 1L, 1L, 1L),
createRLEBlock(0.5, 17));
}
@Test
public void testDoublePartialStep() throws Exception {
// regular approx_percentile
assertAggregation(
DOUBLE_APPROXIMATE_PERCENTILE_AGGREGATION,
1.0,
null,
createDoublesBlock(null, null),
createRLEBlock(0.5, 2));
assertAggregation(
DOUBLE_APPROXIMATE_PERCENTILE_AGGREGATION,
1.0,
1.0,
createDoublesBlock(null, 1.0),
createRLEBlock(0.5, 2));
assertAggregation(
DOUBLE_APPROXIMATE_PERCENTILE_AGGREGATION,
1.0,
2.0,
createDoublesBlock(null, 1.0, 2.0, 3.0),
createRLEBlock(0.5, 4));
assertAggregation(
DOUBLE_APPROXIMATE_PERCENTILE_AGGREGATION,
1.0,
2.0,
createDoublesBlock(1.0, 2.0, 3.0),
createRLEBlock(0.5, 3));
assertAggregation(
DOUBLE_APPROXIMATE_PERCENTILE_AGGREGATION,
1.0,
3.0,
createDoublesBlock(
1.0, null, 2.0, 2.0, null, 2.0, 2.0, null, 2.0, 2.0, null, 3.0, 3.0, null, 3.0, null,
3.0, 4.0, 5.0, 6.0, 7.0),
createRLEBlock(0.5, 21));
// weighted approx_percentile
assertAggregation(
DOUBLE_APPROXIMATE_PERCENTILE_WEIGHTED_AGGREGATION,
1.0,
null,
createDoublesBlock(null, null),
createLongsBlock(1L, 1L),
createRLEBlock(0.5, 2));
assertAggregation(
DOUBLE_APPROXIMATE_PERCENTILE_WEIGHTED_AGGREGATION,
1.0,
1.0,
createDoublesBlock(null, 1.0),
createLongsBlock(1L, 1L),
createRLEBlock(0.5, 2));
assertAggregation(
DOUBLE_APPROXIMATE_PERCENTILE_WEIGHTED_AGGREGATION,
1.0,
2.0,
createDoublesBlock(null, 1.0, 2.0, 3.0),
createLongsBlock(1L, 1L, 1L, 1L),
createRLEBlock(0.5, 4));
assertAggregation(
DOUBLE_APPROXIMATE_PERCENTILE_WEIGHTED_AGGREGATION,
1.0,
2.0,
createDoublesBlock(1.0, 2.0, 3.0),
createLongsBlock(1L, 1L, 1L),
createRLEBlock(0.5, 3));
assertAggregation(
DOUBLE_APPROXIMATE_PERCENTILE_WEIGHTED_AGGREGATION,
1.0,
3.0,
createDoublesBlock(
1.0, null, 2.0, null, 2.0, null, 2.0, null, 3.0, null, 3.0, null, 3.0, 4.0, 5.0, 6.0,
7.0),
createLongsBlock(1L, 1L, 2L, 1L, 2L, 1L, 2L, 1L, 2L, 1L, 1L, 1L, 1L, 1L, 1L, 1L, 1L),
createRLEBlock(0.5, 17));
}
}
