]> Raphaƫl G. Git Repositories - youtubedl/blob - test/test_unicode_literals.py
d/rules: Drop makefile mangling
[youtubedl] / test / test_unicode_literals.py
1 from __future__ import unicode_literals
2
3 # Allow direct execution
4 import os
5 import sys
6 import unittest
7 sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
8
9 import io
10 import re
11
12 rootDir = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
13
14 IGNORED_FILES = [
15 'setup.py', # http://bugs.python.org/issue13943
16 'conf.py',
17 'buildserver.py',
18 ]
19
20 IGNORED_DIRS = [
21 '.git',
22 '.tox',
23 ]
24
25 from test.helper import assertRegexpMatches
26
27
28 class TestUnicodeLiterals(unittest.TestCase):
29 def test_all_files(self):
30 for dirpath, dirnames, filenames in os.walk(rootDir):
31 for ignore_dir in IGNORED_DIRS:
32 if ignore_dir in dirnames:
33 # If we remove the directory from dirnames os.walk won't
34 # recurse into it
35 dirnames.remove(ignore_dir)
36 for basename in filenames:
37 if not basename.endswith('.py'):
38 continue
39 if basename in IGNORED_FILES:
40 continue
41
42 fn = os.path.join(dirpath, basename)
43 with io.open(fn, encoding='utf-8') as inf:
44 code = inf.read()
45
46 if "'" not in code and '"' not in code:
47 continue
48 assertRegexpMatches(
49 self,
50 code,
51 r'(?:(?:#.*?|\s*)\n)*from __future__ import (?:[a-z_]+,\s*)*unicode_literals',
52 'unicode_literals import missing in %s' % fn)
53
54 m = re.search(r'(?<=\s)u[\'"](?!\)|,|$)', code)
55 if m is not None:
56 self.assertTrue(
57 m is None,
58 'u present in %s, around %s' % (
59 fn, code[m.start() - 10:m.end() + 10]))
60
61
62 if __name__ == '__main__':
63 unittest.main()