Search in sources :

Example 6 with Switch

use of com.sri.ai.grinder.sgdpllt.rewriter.core.Switch in project aic-expresso by aic-sri-international.

the class SwitchTest method testConditionalSwitchRewriter.

@Test
public void testConditionalSwitchRewriter() {
    Rewriter baseRewriterSum = e -> new SumStepSolver(e);
    Rewriter baseRewriterSubtraction = e -> new SubtractionStepSolver(e);
    Switch rewriter = new Switch<String>(e -> e.getFunctor() != null ? e.getFunctor().toString() : "", map("+", baseRewriterSum, "-", baseRewriterSubtraction));
    Expression initial;
    Expression expected;
    initial = parse("7");
    expected = parse("7");
    runTest(rewriter, initial, expected, map());
    initial = parse("10 - 3");
    expected = parse("if Inverted then 13 else 7");
    runTest(rewriter, initial, expected, map(inverted, parse("Boolean")));
    initial = parse("10 + 3");
    expected = parse("if Inverted then 7 else 13");
    runTest(rewriter, initial, expected, map(inverted, parse("Boolean")));
}
Also used : ContextSplitting(com.sri.ai.grinder.sgdpllt.core.constraint.ContextSplitting) PropositionalTheory(com.sri.ai.grinder.sgdpllt.theory.propositional.PropositionalTheory) Rewriter(com.sri.ai.grinder.sgdpllt.rewriter.api.Rewriter) DifferenceArithmeticTheory(com.sri.ai.grinder.sgdpllt.theory.differencearithmetic.DifferenceArithmeticTheory) Expression(com.sri.ai.expresso.api.Expression) Solution(com.sri.ai.grinder.sgdpllt.api.ExpressionLiteralSplitterStepSolver.Solution) Recursive(com.sri.ai.grinder.sgdpllt.rewriter.core.Recursive) Util.map(com.sri.ai.util.Util.map) Symbol(com.sri.ai.expresso.api.Symbol) Switch(com.sri.ai.grinder.sgdpllt.rewriter.core.Switch) Expressions.parse(com.sri.ai.expresso.helper.Expressions.parse) Map(java.util.Map) DefaultSymbol(com.sri.ai.expresso.core.DefaultSymbol) DefaultTopRewriter(com.sri.ai.grinder.sgdpllt.rewriter.core.DefaultTopRewriter) TopRewriter(com.sri.ai.grinder.sgdpllt.rewriter.api.TopRewriter) RewriterFromStepMaker(com.sri.ai.grinder.sgdpllt.rewriter.api.RewriterFromStepMaker) Function(com.google.common.base.Function) Util.list(com.sri.ai.util.Util.list) Test(org.junit.Test) Context(com.sri.ai.grinder.sgdpllt.api.Context) List(java.util.List) TrueContext(com.sri.ai.grinder.sgdpllt.core.TrueContext) CompoundTheory(com.sri.ai.grinder.sgdpllt.theory.compound.CompoundTheory) ConstantExpressionStepSolver(com.sri.ai.grinder.sgdpllt.theory.base.ConstantExpressionStepSolver) FirstOf(com.sri.ai.grinder.sgdpllt.rewriter.core.FirstOf) Util(com.sri.ai.util.Util) Assert.assertEquals(org.junit.Assert.assertEquals) ExpressionLiteralSplitterStepSolver(com.sri.ai.grinder.sgdpllt.api.ExpressionLiteralSplitterStepSolver) Switch(com.sri.ai.grinder.sgdpllt.rewriter.core.Switch) Expression(com.sri.ai.expresso.api.Expression) Rewriter(com.sri.ai.grinder.sgdpllt.rewriter.api.Rewriter) DefaultTopRewriter(com.sri.ai.grinder.sgdpllt.rewriter.core.DefaultTopRewriter) TopRewriter(com.sri.ai.grinder.sgdpllt.rewriter.api.TopRewriter) Test(org.junit.Test)

Aggregations

Switch (com.sri.ai.grinder.sgdpllt.rewriter.core.Switch)6 DefaultTopRewriter (com.sri.ai.grinder.sgdpllt.rewriter.core.DefaultTopRewriter)5 Test (org.junit.Test)4 Rewriter (com.sri.ai.grinder.sgdpllt.rewriter.api.Rewriter)3 TopRewriter (com.sri.ai.grinder.sgdpllt.rewriter.api.TopRewriter)3 FirstOf (com.sri.ai.grinder.sgdpllt.rewriter.core.FirstOf)3 Function (com.google.common.base.Function)2 Expression (com.sri.ai.expresso.api.Expression)2 Symbol (com.sri.ai.expresso.api.Symbol)2 DefaultSymbol (com.sri.ai.expresso.core.DefaultSymbol)2 Context (com.sri.ai.grinder.sgdpllt.api.Context)2 Solution (com.sri.ai.grinder.sgdpllt.api.ExpressionLiteralSplitterStepSolver.Solution)2 TrueContext (com.sri.ai.grinder.sgdpllt.core.TrueContext)2 RewriterFromStepMaker (com.sri.ai.grinder.sgdpllt.rewriter.api.RewriterFromStepMaker)2 List (java.util.List)2 Map (java.util.Map)2 Expressions.parse (com.sri.ai.expresso.helper.Expressions.parse)1 ExpressionLiteralSplitterStepSolver (com.sri.ai.grinder.sgdpllt.api.ExpressionLiteralSplitterStepSolver)1 ContextSplitting (com.sri.ai.grinder.sgdpllt.core.constraint.ContextSplitting)1 Recursive (com.sri.ai.grinder.sgdpllt.rewriter.core.Recursive)1