@Test public void testAllColumnsNeeded() { tester().assertThat(new PruneWindowColumns()) .on(p -> buildProjectedWindow(p, alwaysTrue(), alwaysTrue())) .doesNotFire(); }
private RuleAssert assertRuleApplication() { RuleTester tester = tester(); return tester.assertThat(new ExtractSpatialInnerJoin(tester.getMetadata(), tester.getSplitManager(), tester.getPageSourceManager(), tester.getSqlParser())); } }
@Test(expectedExceptions = AssertionError.class, expectedExceptionsMessageRegExp = "Plan does not match, expected .* but found .*") public void testReportWrongMatch() { try (RuleTester tester = new RuleTester()) { tester.assertThat(new DummyReplaceNodeRule()) .on(p -> p.project( Assignments.of(p.symbol("y"), expression("x")), p.values( ImmutableList.of(p.symbol("x")), ImmutableList.of(ImmutableList.of(expression("1")))))) .matches( values(ImmutableList.of("different"), ImmutableList.of())); } }
private RuleAssert assertRuleApplication() { return tester().assertThat(new RewriteSpatialPartitioningAggregation(tester().getMetadata())); } }
@BeforeClass public void setUpBeforeClass() { pickTableLayout = new PickTableLayout(tester().getMetadata(), new SqlParser()); connectorId = tester().getCurrentConnectorId(); nationTableHandle = new TableHandle( connectorId, new TpchTableHandle("nation", 1.0)); ordersTableHandle = new TableHandle( connectorId, new TpchTableHandle("orders", 1.0)); nationTableLayoutHandle = new TableLayoutHandle( connectorId, TestingTransactionHandle.create(), new TpchTableLayoutHandle((TpchTableHandle) nationTableHandle.getConnectorHandle(), TupleDomain.all())); ordersTableLayoutHandle = new TableLayoutHandle( connectorId, TestingTransactionHandle.create(), new TpchTableLayoutHandle((TpchTableHandle) ordersTableHandle.getConnectorHandle(), TupleDomain.all())); }
@BeforeClass public final void setUp() { tester = new RuleTester(plugins); }
@AfterClass(alwaysRun = true) public void tearDown() { tester.close(); tester = null; }
@Test public void doesNotFireOnPlanWithoutApplyNode() { tester().assertThat(new TransformCorrelatedScalarAggregationToJoin(tester().getMetadata().getFunctionRegistry())) .on(p -> p.values(p.symbol("a"))) .doesNotFire(); }
@BeforeClass public void setUp() { tester = new RuleTester(ImmutableList.of(), ImmutableMap.of(), Optional.of(NODES_COUNT)); }
@AfterClass(alwaysRun = true) public void tearDown() { tester.close(); tester = null; }
@Test public void testValueExpressionRewrite() { tester().assertThat(zeroRewriter.valuesExpressionRewrite()) .on(p -> p.values( ImmutableList.<Symbol>of(p.symbol("a")), ImmutableList.of((ImmutableList.of(PlanBuilder.expression("1")))))) .matches( values(ImmutableList.of("a"), ImmutableList.of(ImmutableList.of(new LongLiteral("0"))))); }
private RuleAssert assertRuleApplication() { RuleTester tester = tester(); return tester().assertThat(new ExtractSpatialLeftJoin(tester.getMetadata(), tester.getSplitManager(), tester.getPageSourceManager(), tester.getSqlParser())); } }
@Test public void doesNotFireOnUncorrelated() { tester().assertThat(new TransformCorrelatedScalarAggregationToJoin(tester().getMetadata().getFunctionRegistry())) .on(p -> p.lateral( ImmutableList.of(), p.values(p.symbol("a")), p.values(p.symbol("b")))) .doesNotFire(); }
@BeforeClass public void setUp() { tester = new RuleTester(ImmutableList.of(), ImmutableMap.of(), Optional.of(NODES_COUNT)); }
@Test public void testAllInputsReferenced() { tester().assertThat(new PruneJoinColumns()) .on(p -> buildProjectedJoin(p, Predicates.alwaysTrue())) .doesNotFire(); }
private RuleAssert assertRuleApplication() { return tester().assertThat(new ExtractSpatialInnerJoin(tester().getMetadata(), tester().getSplitManager(), tester().getPageSourceManager())); } }
@Test public void doesNotFireOnCorrelatedWithoutAggregation() { tester().assertThat(new TransformCorrelatedScalarAggregationToJoin(tester().getMetadata().getFunctionRegistry())) .on(p -> p.lateral( ImmutableList.of(p.symbol("corr")), p.values(p.symbol("corr")), p.values(p.symbol("a")))) .doesNotFire(); }
@BeforeClass public void setUp() { tester = new RuleTester( ImmutableList.of(), ImmutableMap.of( JOIN_DISTRIBUTION_TYPE, JoinDistributionType.AUTOMATIC.name(), JOIN_REORDERING_STRATEGY, JoinReorderingStrategy.AUTOMATIC.name()), Optional.of(4)); }
@Test public void testAllInputsReferenced() { tester().assertThat(new PruneCrossJoinColumns()) .on(p -> buildProjectedCrossJoin(p, Predicates.alwaysTrue())) .doesNotFire(); }
private RuleAssert assertRuleApplication() { return tester().assertThat(new ExtractSpatialLeftJoin(tester().getMetadata(), tester().getSplitManager(), tester().getPageSourceManager())); } }