TestDFAMatching.java revision 324c4644fee44b9898524c09511bd33c3f12e2df
1324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver/* 2324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver * [The "BSD license"] 3324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver * Copyright (c) 2010 Terence Parr 4324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver * All rights reserved. 5324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver * 6324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver * Redistribution and use in source and binary forms, with or without 7324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver * modification, are permitted provided that the following conditions 8324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver * are met: 9324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver * 1. Redistributions of source code must retain the above copyright 10324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver * notice, this list of conditions and the following disclaimer. 11324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver * 2. Redistributions in binary form must reproduce the above copyright 12324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver * notice, this list of conditions and the following disclaimer in the 13324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver * documentation and/or other materials provided with the distribution. 14324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver * 3. The name of the author may not be used to endorse or promote products 15324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver * derived from this software without specific prior written permission. 16324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver * 17324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR 18324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES 19324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. 20324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT, 21324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT 22324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, 23324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY 24324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT 25324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF 26324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 27324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver */ 28324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruverpackage org.antlr.test; 29324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver 30324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruverimport org.antlr.analysis.DFA; 31324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruverimport org.antlr.analysis.NFA; 32324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruverimport org.antlr.runtime.ANTLRStringStream; 33324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruverimport org.antlr.tool.Grammar; 34324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruverimport org.junit.Test; 35324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver 36324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruverpublic class TestDFAMatching extends BaseTest { 37324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver 38324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver /** Public default constructor used by TestRig */ 39324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver public TestDFAMatching() { 40324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver } 41324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver 42324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver @Test public void testSimpleAltCharTest() throws Exception { 43324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver Grammar g = new Grammar( 44324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver "lexer grammar t;\n"+ 45324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver "A : {;}'a' | 'b' | 'c';"); 46324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver g.buildNFA(); 47324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver g.createLookaheadDFAs(false); 48324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver DFA dfa = g.getLookaheadDFA(1); 49324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver checkPrediction(dfa,"a",1); 50324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver checkPrediction(dfa,"b",2); 51324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver checkPrediction(dfa,"c",3); 52324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver checkPrediction(dfa,"d", NFA.INVALID_ALT_NUMBER); 53324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver } 54324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver 55324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver @Test public void testSets() throws Exception { 56324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver Grammar g = new Grammar( 57324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver "lexer grammar t;\n"+ 58324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver "A : {;}'a'..'z' | ';' | '0'..'9' ;"); 59324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver g.buildNFA(); 60324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver g.createLookaheadDFAs(false); 61324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver DFA dfa = g.getLookaheadDFA(1); 62324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver checkPrediction(dfa,"a",1); 63324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver checkPrediction(dfa,"q",1); 64324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver checkPrediction(dfa,"z",1); 65324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver checkPrediction(dfa,";",2); 66324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver checkPrediction(dfa,"9",3); 67324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver } 68324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver 69324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver @Test public void testFiniteCommonLeftPrefixes() throws Exception { 70324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver Grammar g = new Grammar( 71324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver "lexer grammar t;\n"+ 72324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver "A : 'a' 'b' | 'a' 'c' | 'd' 'e' ;"); 73324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver g.buildNFA(); 74324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver g.createLookaheadDFAs(false); 75324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver DFA dfa = g.getLookaheadDFA(1); 76324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver checkPrediction(dfa,"ab",1); 77324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver checkPrediction(dfa,"ac",2); 78324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver checkPrediction(dfa,"de",3); 79324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver checkPrediction(dfa,"q", NFA.INVALID_ALT_NUMBER); 80324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver } 81324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver 82324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver @Test public void testSimpleLoops() throws Exception { 83324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver Grammar g = new Grammar( 84324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver "lexer grammar t;\n"+ 85324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver "A : (DIGIT)+ '.' DIGIT | (DIGIT)+ ;\n" + 86324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver "fragment DIGIT : '0'..'9' ;\n"); 87324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver g.buildNFA(); 88324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver g.createLookaheadDFAs(false); 89324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver DFA dfa = g.getLookaheadDFA(3); 90324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver checkPrediction(dfa,"32",2); 91324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver checkPrediction(dfa,"999.2",1); 92324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver checkPrediction(dfa,".2", NFA.INVALID_ALT_NUMBER); 93324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver } 94324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver 95324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver protected void checkPrediction(DFA dfa, String input, int expected) 96324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver throws Exception 97324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver { 98324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver ANTLRStringStream stream = new ANTLRStringStream(input); 99324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver assertEquals(dfa.predict(stream), expected); 100324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver } 101324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver 102324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver} 103