1*16467b97STreehugger Robotimport os 2*16467b97STreehugger Robotimport sys 3*16467b97STreehugger Robotimport antlr3 4*16467b97STreehugger Robotimport testbase 5*16467b97STreehugger Robotimport unittest 6*16467b97STreehugger Robotfrom io import StringIO 7*16467b97STreehugger Robot 8*16467b97STreehugger Robotclass t020fuzzy(testbase.ANTLRTest): 9*16467b97STreehugger Robot def setUp(self): 10*16467b97STreehugger Robot self.compileGrammar('t020fuzzyLexer.g') 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 stream = antlr3.StringStream(f.read()) 17*16467b97STreehugger Robot lexer = self.getLexer(stream) 18*16467b97STreehugger Robot 19*16467b97STreehugger Robot while True: 20*16467b97STreehugger Robot token = lexer.nextToken() 21*16467b97STreehugger Robot if token.type == antlr3.EOF: 22*16467b97STreehugger Robot break 23*16467b97STreehugger Robot 24*16467b97STreehugger Robot 25*16467b97STreehugger Robot output = lexer.output.getvalue() 26*16467b97STreehugger Robot 27*16467b97STreehugger Robot outputPath = os.path.splitext(__file__)[0] + '.output' 28*16467b97STreehugger Robot with open(outputPath) as f: 29*16467b97STreehugger Robot testOutput = f.read() 30*16467b97STreehugger Robot 31*16467b97STreehugger Robot self.assertEqual(output, testOutput) 32*16467b97STreehugger Robot 33*16467b97STreehugger Robot 34*16467b97STreehugger Robotif __name__ == '__main__': 35*16467b97STreehugger Robot unittest.main() 36