2011-10-02 10:29:56 +00:00
|
|
|
package com.sk89q.worldedit.expression;
|
|
|
|
|
|
|
|
import static org.junit.Assert.*;
|
|
|
|
import static java.lang.Math.*;
|
|
|
|
|
|
|
|
import org.junit.*;
|
|
|
|
|
|
|
|
import com.sk89q.worldedit.expression.lexer.LexerException;
|
|
|
|
import com.sk89q.worldedit.expression.parser.ParserException;
|
2011-11-24 19:53:34 +00:00
|
|
|
import com.sk89q.worldedit.expression.runtime.EvaluationException;
|
2011-10-02 10:29:56 +00:00
|
|
|
|
|
|
|
public class ExpressionTest {
|
|
|
|
@Test
|
2011-11-22 03:11:58 +00:00
|
|
|
public void testEvaluate() throws ExpressionException {
|
2011-10-02 10:29:56 +00:00
|
|
|
// check
|
2011-11-23 01:29:48 +00:00
|
|
|
assertEquals(1 - 2 + 3, simpleEval("1 - 2 + 3"), 0);
|
2011-10-02 10:29:56 +00:00
|
|
|
|
|
|
|
// check unary ops
|
2011-11-23 01:29:48 +00:00
|
|
|
assertEquals(2 + +4, simpleEval("2 + +4"), 0);
|
|
|
|
assertEquals(2 - -4, simpleEval("2 - -4"), 0);
|
|
|
|
assertEquals(2 * -4, simpleEval("2 * -4"), 0);
|
2011-10-02 10:29:56 +00:00
|
|
|
|
|
|
|
// check functions
|
|
|
|
assertEquals(sin(5), simpleEval("sin(5)"), 0);
|
2011-11-23 01:29:48 +00:00
|
|
|
assertEquals(atan2(3, 4), simpleEval("atan2(3, 4)"), 0);
|
2011-10-02 10:29:56 +00:00
|
|
|
|
|
|
|
// check variables
|
2011-11-24 19:53:34 +00:00
|
|
|
assertEquals(8, compile("foo+bar", "foo", "bar").evaluate(5, 3), 0);
|
2011-10-02 10:29:56 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
@Test
|
|
|
|
public void testErrors() throws ExpressionException {
|
|
|
|
// test lexer errors
|
|
|
|
try {
|
2011-11-24 19:53:34 +00:00
|
|
|
compile("#");
|
2011-10-02 10:29:56 +00:00
|
|
|
fail("Error expected");
|
|
|
|
} catch (LexerException e) {
|
|
|
|
assertEquals("Error position", 0, e.getPosition());
|
|
|
|
}
|
|
|
|
|
|
|
|
// test parser errors
|
|
|
|
try {
|
2011-11-24 19:53:34 +00:00
|
|
|
compile("x");
|
2011-10-02 10:29:56 +00:00
|
|
|
fail("Error expected");
|
|
|
|
} catch (ParserException e) {
|
|
|
|
assertEquals("Error position", 0, e.getPosition());
|
|
|
|
}
|
|
|
|
try {
|
2011-11-24 19:53:34 +00:00
|
|
|
compile("x()");
|
2011-10-02 10:29:56 +00:00
|
|
|
fail("Error expected");
|
|
|
|
} catch (ParserException e) {
|
|
|
|
assertEquals("Error position", 0, e.getPosition());
|
|
|
|
}
|
|
|
|
try {
|
2011-11-24 19:53:34 +00:00
|
|
|
compile("(");
|
2011-10-02 10:29:56 +00:00
|
|
|
fail("Error expected");
|
|
|
|
} catch (ParserException e) {}
|
|
|
|
try {
|
2011-11-24 19:53:34 +00:00
|
|
|
compile("x(");
|
2011-10-02 10:29:56 +00:00
|
|
|
fail("Error expected");
|
|
|
|
} catch (ParserException e) {}
|
|
|
|
}
|
|
|
|
|
2011-10-23 03:15:23 +00:00
|
|
|
@Test
|
|
|
|
public void testAssign() throws ExpressionException {
|
2011-11-24 19:53:34 +00:00
|
|
|
Expression foo = compile("{a=x} b=y; c=z", "x", "y", "z", "a", "b", "c");
|
2011-10-23 03:15:23 +00:00
|
|
|
foo.evaluate(2, 3, 5);
|
2011-10-29 14:03:55 +00:00
|
|
|
assertEquals(2, foo.getVariable("a").getValue(), 0);
|
|
|
|
assertEquals(3, foo.getVariable("b").getValue(), 0);
|
|
|
|
assertEquals(5, foo.getVariable("c").getValue(), 0);
|
2011-10-23 03:15:23 +00:00
|
|
|
}
|
|
|
|
|
2011-11-22 03:11:58 +00:00
|
|
|
@Test
|
|
|
|
public void testIf() throws ExpressionException {
|
|
|
|
assertEquals(40, simpleEval("if (1) x=4; else y=5; x*10+y;"), 0);
|
|
|
|
assertEquals(5, simpleEval("if (0) x=4; else y=5; x*10+y;"), 0);
|
|
|
|
|
|
|
|
// test 'dangling else'
|
2011-11-24 19:53:34 +00:00
|
|
|
final Expression expression1 = compile("if (1) if (0) x=4; else y=5;", "x", "y");
|
2011-11-22 03:11:58 +00:00
|
|
|
expression1.evaluate(1, 2);
|
|
|
|
assertEquals(1, expression1.getVariable("x").getValue(), 0);
|
|
|
|
assertEquals(5, expression1.getVariable("y").getValue(), 0);
|
|
|
|
|
|
|
|
// test if the if construct is correctly recognized as a statement
|
2011-11-24 19:53:34 +00:00
|
|
|
final Expression expression2 = compile("if (0) if (1) x=5; y=4;", "x", "y");
|
2011-11-22 03:11:58 +00:00
|
|
|
expression2.evaluate(1, 2);
|
|
|
|
assertEquals(4, expression2.getVariable("y").getValue(), 0);
|
2011-11-22 05:07:22 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
@Test
|
|
|
|
public void testWhile() throws ExpressionException {
|
|
|
|
assertEquals(5, simpleEval("c=5; a=0; while (c > 0) { ++a; --c; } a"), 0);
|
|
|
|
assertEquals(5, simpleEval("c=5; a=0; do { ++a; --c; } while (c > 0) a"), 0);
|
2011-11-22 03:11:58 +00:00
|
|
|
}
|
|
|
|
|
2011-11-22 15:08:15 +00:00
|
|
|
@Test
|
|
|
|
public void testFor() throws ExpressionException {
|
|
|
|
assertEquals(5, simpleEval("a=0; for (i=0; i<5; ++i) { ++a; } a"), 0);
|
2011-11-24 01:32:55 +00:00
|
|
|
assertEquals(12345, simpleEval("y=0; for (i=1,5) { y *= 10; y += i; } y"), 0);
|
2011-11-22 15:08:15 +00:00
|
|
|
}
|
|
|
|
|
2011-11-24 19:53:34 +00:00
|
|
|
private double simpleEval(String expressionString) throws ExpressionException {
|
|
|
|
final Expression expression = compile(expressionString);
|
|
|
|
return expression.evaluate();
|
|
|
|
}
|
|
|
|
|
|
|
|
private Expression compile(String expressionString, String... variableNames) throws ExpressionException, EvaluationException {
|
|
|
|
final Expression expression = Expression.compile(expressionString, variableNames);
|
|
|
|
expression.optimize();
|
|
|
|
return expression;
|
2011-10-02 10:29:56 +00:00
|
|
|
}
|
|
|
|
}
|