use of com.facebook.presto.operator.TopNOperator.TopNOperatorFactory in project presto by prestodb.
the class TestTopNOperator method testExceedMemoryLimit.
@Test
public void testExceedMemoryLimit() throws Exception {
List<Page> input = rowPagesBuilder(BIGINT).row(1L).build();
DriverContext smallDiverContext = createTaskContext(executor, scheduledExecutor, TEST_SESSION, new DataSize(1, BYTE)).addPipelineContext(0, true, true, false).addDriverContext();
TopNOperatorFactory operatorFactory = new TopNOperatorFactory(0, new PlanNodeId("test"), ImmutableList.of(BIGINT), 100, ImmutableList.of(0), ImmutableList.of(ASC_NULLS_LAST));
try (Operator operator = operatorFactory.createOperator(smallDiverContext)) {
operator.addInput(input.get(0));
fail("must fail because of exceeding local memory limit");
} catch (ExceededMemoryLimitException ignore) {
}
}
use of com.facebook.presto.operator.TopNOperator.TopNOperatorFactory in project presto by prestodb.
the class TestTopNOperator method testReverseOrder.
@Test
public void testReverseOrder() {
List<Page> input = rowPagesBuilder(BIGINT, DOUBLE).row(1L, 0.1).row(2L, 0.2).pageBreak().row(-1L, -0.1).row(4L, 0.4).pageBreak().row(5L, 0.5).row(4L, 0.41).row(6L, 0.6).pageBreak().build();
TopNOperatorFactory operatorFactory = new TopNOperatorFactory(0, new PlanNodeId("test"), ImmutableList.of(BIGINT, DOUBLE), 2, ImmutableList.of(0), ImmutableList.of(ASC_NULLS_LAST));
MaterializedResult expected = resultBuilder(driverContext.getSession(), BIGINT, DOUBLE).row(-1L, -0.1).row(1L, 0.1).build();
assertOperatorEquals(operatorFactory, driverContext, input, expected);
}
Aggregations