1*16467b97STreehugger Robotimport antlr3 2*16467b97STreehugger Robotimport testbase 3*16467b97STreehugger Robotimport unittest 4*16467b97STreehugger Robot 5*16467b97STreehugger Robot 6*16467b97STreehugger Robotclass t037rulePropertyRef(testbase.ANTLRTest): 7*16467b97STreehugger Robot def setUp(self): 8*16467b97STreehugger Robot self.compileGrammar() 9*16467b97STreehugger Robot 10*16467b97STreehugger Robot 11*16467b97STreehugger Robot def lexerClass(self, base): 12*16467b97STreehugger Robot class TLexer(base): 13*16467b97STreehugger Robot def recover(self, input, re): 14*16467b97STreehugger Robot # no error recovery yet, just crash! 15*16467b97STreehugger Robot raise 16*16467b97STreehugger Robot 17*16467b97STreehugger Robot return TLexer 18*16467b97STreehugger Robot 19*16467b97STreehugger Robot 20*16467b97STreehugger Robot def parserClass(self, base): 21*16467b97STreehugger Robot class TParser(base): 22*16467b97STreehugger Robot def recover(self, input, re): 23*16467b97STreehugger Robot # no error recovery yet, just crash! 24*16467b97STreehugger Robot raise 25*16467b97STreehugger Robot 26*16467b97STreehugger Robot return TParser 27*16467b97STreehugger Robot 28*16467b97STreehugger Robot 29*16467b97STreehugger Robot def testValid1(self): 30*16467b97STreehugger Robot cStream = antlr3.StringStream(' a a a a ') 31*16467b97STreehugger Robot 32*16467b97STreehugger Robot lexer = self.getLexer(cStream) 33*16467b97STreehugger Robot tStream = antlr3.CommonTokenStream(lexer) 34*16467b97STreehugger Robot parser = self.getParser(tStream) 35*16467b97STreehugger Robot start, stop, text = parser.a().bla 36*16467b97STreehugger Robot 37*16467b97STreehugger Robot # first token of rule b is the 2nd token (counting hidden tokens) 38*16467b97STreehugger Robot assert start.index == 1, start 39*16467b97STreehugger Robot 40*16467b97STreehugger Robot # first token of rule b is the 7th token (counting hidden tokens) 41*16467b97STreehugger Robot assert stop.index == 7, stop 42*16467b97STreehugger Robot 43*16467b97STreehugger Robot assert text == "a a a a", text 44*16467b97STreehugger Robot 45*16467b97STreehugger Robot 46*16467b97STreehugger Robotif __name__ == '__main__': 47*16467b97STreehugger Robot unittest.main() 48