xref: /aosp_15_r20/external/antlr/runtime/Python3/tests/t020fuzzy.py (revision 16467b971bd3e2009fad32dd79016f2c7e421deb)
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