initial commit

java/expression/parser/ParserTester.java

@@ -0,0 +1,76 @@
+package expression.parser;
+
+import base.ExtendedRandom;
+import base.TestCounter;
+import base.Tester;
+import base.Unit;
+import expression.ToMiniString;
+import expression.common.*;
+
+import java.util.ArrayList;
+import java.util.List;
+import java.util.function.BiFunction;
+import java.util.function.LongBinaryOperator;
+import java.util.function.LongToIntFunction;
+
+/**
+ * @author Georgiy Korneev (kgeorgiy@kgeorgiy.info)
+ */
+public class ParserTester extends Tester {
+    /* package-private */ final TestGeneratorBuilder<Integer> generator;
+    /* package-private */ final Renderer.Builder<Integer, Unit, ParserTestSet.TExpression> renderer;
+    private final List<ParserTestSet.ParsedKind<?, ?>> kinds = new ArrayList<>();
+    /* package-private */  final List<NodeRenderer.Paren> parens = new ArrayList<>(List.of(NodeRenderer.paren("(", ")")));
+
+    public ParserTester(final TestCounter counter) {
+        super(counter);
+        renderer = Renderer.builder(c -> vars -> c);
+        final ExtendedRandom random = counter.random();
+        generator = new TestGeneratorBuilder<>(random, random::nextInt, ParserTestSet.CONSTS, true);
+    }
+
+    public void unary(final String name, final int priority, final BiFunction<Long, LongToIntFunction, Long> op) {
+        generator.unary(name, priority);
+        renderer.unary(name, (unit, a) -> vars -> cast(op.apply(a.evaluate(vars), this::cast)));
+    }
+
+    public void unary(final String left, final String right, final BiFunction<Long, LongToIntFunction, Long> op) {
+        generator.unary(left, right);
+        renderer.unary(left, (unit, a) -> vars -> cast(op.apply(a.evaluate(vars), this::cast)));
+    }
+
+    public void binary(final String name, final int priority, final LongBinaryOperator op) {
+        generator.binary(name, priority);
+        renderer.binary(name, (unit, a, b) -> vars -> cast(op.applyAsLong(a.evaluate(vars), b.evaluate(vars))));
+    }
+
+    <E extends ToMiniString, C> void kind(final ExpressionKind<E, C> kind, final ParserTestSet.Parser<E> parser) {
+        kinds.add(new ParserTestSet.ParsedKind<>(kind, parser));
+    }
+
+    @Override
+    public void test() {
+        for (final ParserTestSet.ParsedKind<?, ?> kind : kinds) {
+            counter.scope(kind.toString(), () -> test(kind));
+        }
+    }
+
+    protected void test(final ParserTestSet.ParsedKind<?, ?> kind) {
+        new ParserTestSet<>(this, kind).test();
+    }
+
+    public TestCounter getCounter() {
+        return counter;
+    }
+
+    protected int cast(final long value) {
+        return (int) value;
+    }
+
+    public void parens(final String... parens) {
+        assert parens.length % 2 == 0 : "Parens should come in pairs";
+        for (int i = 0; i < parens.length; i += 2) {
+            this.parens.add(NodeRenderer.paren(parens[i], parens[i + 1]));
+        }
+    }
+}