1*16467b97STreehugger Robotimport os 2*16467b97STreehugger Robotimport sys 3*16467b97STreehugger Robotimport antlr3 4*16467b97STreehugger Robotimport testbase 5*16467b97STreehugger Robotimport unittest 6*16467b97STreehugger Robot 7*16467b97STreehugger Robot 8*16467b97STreehugger Robotclass t021hoist(testbase.ANTLRTest): 9*16467b97STreehugger Robot def setUp(self): 10*16467b97STreehugger Robot self.compileGrammar() 11*16467b97STreehugger Robot 12*16467b97STreehugger Robot 13*16467b97STreehugger Robot def testValid1(self): 14*16467b97STreehugger Robot cStream = antlr3.StringStream('enum') 15*16467b97STreehugger Robot lexer = self.getLexer(cStream) 16*16467b97STreehugger Robot tStream = antlr3.CommonTokenStream(lexer) 17*16467b97STreehugger Robot parser = self.getParser(tStream) 18*16467b97STreehugger Robot parser.enableEnum = True 19*16467b97STreehugger Robot enumIs = parser.stat() 20*16467b97STreehugger Robot 21*16467b97STreehugger Robot assert enumIs == 'keyword', repr(enumIs) 22*16467b97STreehugger Robot 23*16467b97STreehugger Robot 24*16467b97STreehugger Robot def testValid2(self): 25*16467b97STreehugger Robot cStream = antlr3.StringStream('enum') 26*16467b97STreehugger Robot lexer = self.getLexer(cStream) 27*16467b97STreehugger Robot tStream = antlr3.CommonTokenStream(lexer) 28*16467b97STreehugger Robot parser = self.getParser(tStream) 29*16467b97STreehugger Robot parser.enableEnum = False 30*16467b97STreehugger Robot enumIs = parser.stat() 31*16467b97STreehugger Robot 32*16467b97STreehugger Robot assert enumIs == 'ID', repr(enumIs) 33*16467b97STreehugger Robot 34*16467b97STreehugger Robot 35*16467b97STreehugger Robot 36*16467b97STreehugger Robotif __name__ == '__main__': 37*16467b97STreehugger Robot unittest.main() 38*16467b97STreehugger Robot 39