1*16467b97STreehugger Robotimport antlr3 2*16467b97STreehugger Robotimport testbase 3*16467b97STreehugger Robotimport unittest 4*16467b97STreehugger Robotimport os 5*16467b97STreehugger Robotimport sys 6*16467b97STreehugger Robotfrom io import StringIO 7*16467b97STreehugger Robot 8*16467b97STreehugger Robotclass t018llstar(testbase.ANTLRTest): 9*16467b97STreehugger Robot def setUp(self): 10*16467b97STreehugger Robot self.compileGrammar() 11*16467b97STreehugger Robot 12*16467b97STreehugger Robot 13*16467b97STreehugger Robot def testValid(self): 14*16467b97STreehugger Robot inputPath = os.path.splitext(__file__)[0] + '.input' 15*16467b97STreehugger Robot with open(inputPath) as f: 16*16467b97STreehugger Robot cStream = antlr3.StringStream(f.read()) 17*16467b97STreehugger Robot lexer = self.getLexer(cStream) 18*16467b97STreehugger Robot tStream = antlr3.CommonTokenStream(lexer) 19*16467b97STreehugger Robot parser = self.getParser(tStream) 20*16467b97STreehugger Robot parser.program() 21*16467b97STreehugger Robot 22*16467b97STreehugger Robot output = parser.output.getvalue() 23*16467b97STreehugger Robot 24*16467b97STreehugger Robot outputPath = os.path.splitext(__file__)[0] + '.output' 25*16467b97STreehugger Robot with open(outputPath) as f: 26*16467b97STreehugger Robot testOutput = f.read() 27*16467b97STreehugger Robot 28*16467b97STreehugger Robot self.assertEqual(output, testOutput) 29*16467b97STreehugger Robot 30*16467b97STreehugger Robotif __name__ == '__main__': 31*16467b97STreehugger Robot unittest.main() 32