diff options
Diffstat (limited to 'tests/test_examplefiles.py')
| -rw-r--r-- | tests/test_examplefiles.py | 17 |
1 files changed, 9 insertions, 8 deletions
diff --git a/tests/test_examplefiles.py b/tests/test_examplefiles.py index eabb0cf7..12f9d62f 100644 --- a/tests/test_examplefiles.py +++ b/tests/test_examplefiles.py @@ -22,20 +22,21 @@ STORE_OUTPUT = False STATS = {} +TESTDIR = os.path.dirname(__file__) + # generate methods def test_example_files(): global STATS STATS = {} - testdir = os.path.dirname(__file__) - outdir = os.path.join(testdir, 'examplefiles', 'output') + outdir = os.path.join(TESTDIR, 'examplefiles', 'output') if STORE_OUTPUT and not os.path.isdir(outdir): os.makedirs(outdir) - for fn in os.listdir(os.path.join(testdir, 'examplefiles')): + for fn in os.listdir(os.path.join(TESTDIR, 'examplefiles')): if fn.startswith('.') or fn.endswith('#'): continue - absfn = os.path.join(testdir, 'examplefiles', fn) + absfn = os.path.join(TESTDIR, 'examplefiles', fn) if not os.path.isfile(absfn): continue @@ -46,8 +47,6 @@ def test_example_files(): except UnicodeError: code = code.decode('latin1') - outfn = os.path.join(outdir, fn) - lx = None if '_' in fn: try: @@ -62,7 +61,7 @@ def test_example_files(): 'nor is of the form <lexer>_filename ' 'for overriding, thus no lexer found.' % fn) - yield check_lexer, lx, absfn, outfn + yield check_lexer, lx, fn N = 7 stats = list(STATS.items()) @@ -77,7 +76,8 @@ def test_example_files(): print('%-30s %6d chars %8.2f ms %7.3f ms/char' % ((fn,) + t)) -def check_lexer(lx, absfn, outfn): +def check_lexer(lx, fn): + absfn = os.path.join(TESTDIR, 'examplefiles', fn) fp = open(absfn, 'rb') try: text = fp.read() @@ -112,6 +112,7 @@ def check_lexer(lx, absfn, outfn): # check output against previous run if enabled if STORE_OUTPUT: # no previous output -- store it + outfn = os.path.join(TESTDIR, 'examplefiles', 'output', fn) if not os.path.isfile(outfn): fp = open(outfn, 'wb') try: |