Raphaël G. Git Repositories - youtubedl/blob - 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()