Examples of org.apache.pig.newplan.logical.expression.SubtractExpression

        ConstantExpression constant1 = new ConstantExpression(plan, 10) ;
        ConstantExpression constant2 =  new ConstantExpression(plan,  new DataByteArray()) ;
        ConstantExpression constant3 =  new ConstantExpression(plan,  123L) ;
        ConstantExpression constant4 =  new ConstantExpression(plan,  true) ;


        SubtractExpression sub1 = new SubtractExpression(plan, constant1, constant2) ;
        GreaterThanExpression gt1 = new GreaterThanExpression(plan, sub1, constant3) ;
        AndExpression and1 = new AndExpression(plan, gt1, constant4) ;
        NotExpression not1 = new NotExpression(plan, and1) ;


        CompilationMessageCollector collector = new CompilationMessageCollector() ;
        TypeCheckingExpVisitor expTypeChecker = new TypeCheckingExpVisitor(plan, collector, null);
        expTypeChecker.visit();




        printMessageCollector(collector) ;
        //printTypeGraph(plan) ;


        if (collector.hasError()) {
            throw new Exception("Error not expected during type checking") ;
        }       




        // Induction check   
        assertEquals(DataType.INTEGER, sub1.getType()) ;    
        assertEquals(DataType.BOOLEAN, gt1.getType()) ;     
        assertEquals(DataType.BOOLEAN, and1.getType()) ;    
        assertEquals(DataType.BOOLEAN, not1.getType()) ;    


        // Cast insertion check     
        assertEquals(DataType.INTEGER, sub1.getRhs().getType()) ;    
        assertEquals(DataType.LONG, gt1.getLhs().getType()) ;


    }

        LogicalExpressionPlan plan = new LogicalExpressionPlan() ;
        ConstantExpression constant1 = new ConstantExpression(plan, 10) ;
        ConstantExpression constant2 =  new ConstantExpression(plan, 20L) ;


        NegativeExpression neg1 = new NegativeExpression(plan, constant1) ;
        SubtractExpression subtract1 = new SubtractExpression(plan, neg1, constant2) ;


        // Before type checking its set correctly = PIG-421
//        assertEquals(DataType.LONG, subtract1.getType()) ;


        CompilationMessageCollector collector = new CompilationMessageCollector() ;
        TypeCheckingExpVisitor expTypeChecker = new TypeCheckingExpVisitor(plan, collector, null);
        expTypeChecker.visit();   
        printMessageCollector(collector) ;


        //printTypeGraph(plan) ;


        // After type checking
        System.out.println(DataType.findTypeName(subtract1.getType())) ;
        assertEquals(DataType.LONG, subtract1.getType()) ;


        assertTrue(subtract1.getLhs() instanceof CastExpression);
        assertEquals(((CastExpression)subtract1.getLhs()).getType(), DataType.LONG);
        assertTrue(((CastExpression)subtract1.getLhs()).getExpression() == neg1);
    }

Examples of org.apache.pig.newplan.logical.expression.SubtractExpression

Related Classes of org.apache.pig.newplan.logical.expression.SubtractExpression