diff --git a/librefi/fxckers/__init__.py b/librefi/fxckers/__init__.py index d642f2d..2a98d9c 100644 --- a/librefi/fxckers/__init__.py +++ b/librefi/fxckers/__init__.py @@ -1,3 +1,6 @@ +# -*- coding: utf-8 -*- +from __future__ import unicode_literals + from ._map import fxckers_map __all__ = ['fxckers_map'] diff --git a/tests/py2_compat.py b/tests/py2_compat.py new file mode 100644 index 0000000..dd8f18a --- /dev/null +++ b/tests/py2_compat.py @@ -0,0 +1,19 @@ +# -*- coding: utf-8 -*- +from __future__ import unicode_literals + +from os import listdir, path +import re + + +def test_fxcker_files_py2_compatibility(): + dir = path.join('.', 'librefi', 'fxckers') + fxckers = listdir(dir) + for fxcker in fxckers: + if re.match(r'^\w+\.py$', fxcker): + with open(path.join(dir, fxcker), 'r') as file: + content = file.read() + if re.match(r'^(#!.+\r?\n)?\s*(# (-\*- )?coding: utf-8( -\*-)?\r?\n)?\s*(from __future__ import (\w+, )*unicode_literals)', + content) is None: + print(content) + print(fxcker) + raise AssertionError('fxcker %s not containing required compat imports' % (fxcker))