64 lines
1.8 KiB
Python
64 lines
1.8 KiB
Python
from __future__ import unicode_literals
|
|
|
|
# Allow direct execution
|
|
import os
|
|
import sys
|
|
import unittest
|
|
sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
|
|
|
|
import io
|
|
import re
|
|
|
|
rootDir = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
|
|
|
|
IGNORED_FILES = [
|
|
'setup.py', # http://bugs.python.org/issue13943
|
|
'conf.py',
|
|
'buildserver.py',
|
|
]
|
|
|
|
IGNORED_DIRS = [
|
|
'.git',
|
|
'.tox',
|
|
]
|
|
|
|
from test.helper import assertRegexpMatches
|
|
|
|
|
|
class TestUnicodeLiterals(unittest.TestCase):
|
|
def test_all_files(self):
|
|
for dirpath, dirnames, filenames in os.walk(rootDir):
|
|
for ignore_dir in IGNORED_DIRS:
|
|
if ignore_dir in dirnames:
|
|
# If we remove the directory from dirnames os.walk won't
|
|
# recurse into it
|
|
dirnames.remove(ignore_dir)
|
|
for basename in filenames:
|
|
if not basename.endswith('.py'):
|
|
continue
|
|
if basename in IGNORED_FILES:
|
|
continue
|
|
|
|
fn = os.path.join(dirpath, basename)
|
|
with io.open(fn, encoding='utf-8') as inf:
|
|
code = inf.read()
|
|
|
|
if "'" not in code and '"' not in code:
|
|
continue
|
|
assertRegexpMatches(
|
|
self,
|
|
code,
|
|
r'(?:(?:#.*?|\s*)\n)*from __future__ import (?:[a-z_]+,\s*)*unicode_literals',
|
|
'unicode_literals import missing in %s' % fn)
|
|
|
|
m = re.search(r'(?<=\s)u[\'"](?!\)|,|$)', code)
|
|
if m is not None:
|
|
self.assertTrue(
|
|
m is None,
|
|
'u present in %s, around %s' % (
|
|
fn, code[m.start() - 10:m.end() + 10]))
|
|
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|