예제 #1
0
  @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));
  }
}