Merge pull request #840 from ZyX-I/review-res-fixes

Fixes #819.

powerline/bindings/vim/__init__.py

@@ -129,6 +129,7 @@ if sys.version_info < (3,):
 		return buf.name
 else:
 	vim_bufname = vim_get_func('bufname')
+
 	def buffer_name(buf):  # NOQA
 		try:
 			name = buf.name

powerline/lib/file_watcher.py

@@ -13,11 +13,12 @@ from threading import RLock
 from powerline.lib.monotonic import monotonic
 from powerline.lib.inotify import INotify, INotifyError
 
+
 def realpath(path):
 	return os.path.abspath(os.path.realpath(path))
 
-class INotifyWatch(INotify):
 
+class INotifyWatch(INotify):
 	is_stat_based = False
 
 	def __init__(self, expire_time=10):

powerline/lib/humanize_bytes.py

@@ -17,6 +17,6 @@ def humanize_bytes(num, suffix='B', si_prefix=False):
 	unit, decimals = unit_list[exponent]
 	if unit and not si_prefix:
 		unit = unit.upper() + 'i'
-	return '{{quotient:.{decimals}f}} {{unit}}{{suffix}}'\
+	return ('{{quotient:.{decimals}f}} {{unit}}{{suffix}}'
-		.format(decimals=decimals)\
+		.format(decimals=decimals)
-		.format(quotient=quotient, unit=unit, suffix=suffix)
+		.format(quotient=quotient, unit=unit, suffix=suffix))

powerline/lib/tree_watcher.py

@@ -16,19 +16,21 @@ from powerline.lib.inotify import INotify, INotifyError
 class NoSuchDir(ValueError):
 	pass
 
+
 class BaseDirChanged(ValueError):
 	pass
 
-class DirTooLarge(ValueError):
 
+class DirTooLarge(ValueError):
 	def __init__(self, bdir):
 		ValueError.__init__(self, 'The directory {0} is too large to monitor. Try increasing the value in /proc/sys/fs/inotify/max_user_watches'.format(bdir))
 
+
 def realpath(path):
 	return os.path.abspath(os.path.realpath(path))
 
-class INotifyTreeWatcher(INotify):
 
+class INotifyTreeWatcher(INotify):
 	is_dummy = False
 
 	def __init__(self, basedir, ignore_event=None):
@@ -141,7 +143,6 @@ class INotifyTreeWatcher(INotify):
 
 
 class DummyTreeWatcher(object):
-
 	is_dummy = True
 
 	def __init__(self, basedir):
@@ -150,8 +151,8 @@ class DummyTreeWatcher(object):
 	def __call__(self):
 		return False
 
-class TreeWatcher(object):
 
+class TreeWatcher(object):
 	def __init__(self, expire_time=10):
 		self.watches = {}
 		self.last_query_times = {}
@@ -203,6 +204,7 @@ class TreeWatcher(object):
 			self.watches[path] = DummyTreeWatcher(path)
 			return False
 
+
 if __name__ == '__main__':
 	w = INotifyTreeWatcher(sys.argv[-1])
 	w()

powerline/lib/vcs/__init__.py

@@ -1,16 +1,11 @@
 # vim:fileencoding=utf-8:noet
 from __future__ import absolute_import
 
-import os, errno
+import os
+import errno
 from threading import Lock
 from collections import defaultdict
 
-vcs_props = (
-	('git', '.git', os.path.exists),
-	('mercurial', '.hg', os.path.isdir),
-	('bzr', '.bzr', os.path.isdir),
-)
-
 
 def generate_directories(path):
 	if os.path.isdir(path):
@@ -24,8 +19,10 @@ def generate_directories(path):
 			break
 		yield path
 
+
 _file_watcher = None
 
+
 def file_watcher():
 	global _file_watcher
 	if _file_watcher is None:
@@ -33,8 +30,10 @@ def file_watcher():
 		_file_watcher = create_file_watcher()
 	return _file_watcher
 
+
 _branch_watcher = None
 
+
 def branch_watcher():
 	global _branch_watcher
 	if _branch_watcher is None:
@@ -42,10 +41,12 @@ def branch_watcher():
 		_branch_watcher = create_file_watcher()
 	return _branch_watcher
 
+
 branch_name_cache = {}
 branch_lock = Lock()
 file_status_lock = Lock()
 
+
 def get_branch_name(directory, config_file, get_func):
 	global branch_name_cache
 	with branch_lock:
@@ -79,8 +80,8 @@ def get_branch_name(directory, config_file, get_func):
 			branch_name_cache[config_file] = get_func(directory, config_file)
 		return branch_name_cache[config_file]
 
-class FileStatusCache(dict):
 
+class FileStatusCache(dict):
 	def __init__(self):
 		self.dirstate_map = defaultdict(set)
 		self.ignore_map = defaultdict(set)
@@ -112,8 +113,10 @@ class FileStatusCache(dict):
 		for ignf in self.keypath_ignore_map[keypath]:
 			yield ignf
 
+
 file_status_cache = FileStatusCache()
 
+
 def get_file_status(directory, dirstate_file, file_path, ignore_file_name, get_func, extra_ignore_files=()):
 	global file_status_cache
 	keypath = file_path if os.path.isabs(file_path) else os.path.join(directory, file_path)
@@ -175,8 +178,8 @@ def get_file_status(directory, dirstate_file, file_path, ignore_file_name, get_f
 			file_status_cache[keypath] = ans = get_func(directory, file_path)
 			return ans
 
-class TreeStatusCache(dict):
 
+class TreeStatusCache(dict):
 	def __init__(self):
 		from powerline.lib.tree_watcher import TreeWatcher
 		self.tw = TreeWatcher()
@@ -196,14 +199,24 @@ class TreeStatusCache(dict):
 			logger.warn('Failed to check %s for changes, with error: %s'% key, e)
 		return self.cache_and_get(key, repo.status)
 
+
 _tree_status_cache = None
 
+
 def tree_status(repo, logger):
 	global _tree_status_cache
 	if _tree_status_cache is None:
 		_tree_status_cache = TreeStatusCache()
 	return _tree_status_cache(repo, logger)
 
+
+vcs_props = (
+	('git', '.git', os.path.exists),
+	('mercurial', '.hg', os.path.isdir),
+	('bzr', '.bzr', os.path.isdir),
+)
+
+
 def guess(path):
 	for directory in generate_directories(path):
 		for vcs, vcs_dir, check in vcs_props:
@@ -219,8 +232,12 @@ def guess(path):
 					pass
 	return None
 
+
 def debug():
-	''' To use run python -c "from powerline.lib.vcs import debug; debug()" some_file_to_watch '''
+	'''Test run guess(), repo.branch() and repo.status()
+
+	To use run python -c "from powerline.lib.vcs import debug; debug()" 
+	some_file_to_watch '''
 	import sys
 	dest = sys.argv[-1]
 	repo = guess(os.path.abspath(dest))

powerline/lib/vcs/bzr.py

@@ -10,16 +10,17 @@ from bzrlib import (workingtree, status, library_state, trace, ui)
 
 from powerline.lib.vcs import get_branch_name, get_file_status
 
+
 class CoerceIO(StringIO):
 	def write(self, arg):
 		if isinstance(arg, bytes):
 			arg = arg.decode('utf-8', 'replace')
 		return super(CoerceIO, self).write(arg)
 
-state = None
 
 nick_pat = re.compile(br'nickname\s*=\s*(.+)')
 
+
 def branch_name_from_config_file(directory, config_file):
 	ans = None
 	try:
@@ -33,8 +34,11 @@ def branch_name_from_config_file(directory, config_file):
 		pass
 	return ans or os.path.basename(directory)
 
-class Repository(object):
 
+state = None
+
+
+class Repository(object):
 	def __init__(self, directory):
 		if isinstance(directory, bytes):
 			directory = directory.decode(sys.getfilesystemencoding() or sys.getdefaultencoding() or 'utf-8')
@@ -75,7 +79,7 @@ class Repository(object):
 			return
 		if path:
 			ans = raw[:2]
-			if ans == 'I ': # Ignored
+			if ans == 'I ':  # Ignored
 				ans = None
 			return ans
 		dirtied = untracked = ' '
@@ -90,4 +94,3 @@ class Repository(object):
 	def branch(self):
 		config_file = os.path.join(self.directory, '.bzr', 'branch', 'branch.conf')
 		return get_branch_name(self.directory, config_file, branch_name_from_config_file)
-

powerline/lib/vcs/git.py

@@ -2,12 +2,13 @@
 
 import os
 import re
-import errno
 
 from powerline.lib.vcs import get_branch_name as _get_branch_name, get_file_status
 
+
 _ref_pat = re.compile(br'ref:\s*refs/heads/(.+)')
 
+
 def branch_name_from_config_file(directory, config_file):
 	try:
 		with open(config_file, 'rb') as f:
@@ -19,6 +20,7 @@ def branch_name_from_config_file(directory, config_file):
 		return m.group(1).decode('utf-8', 'replace')
 	return raw[:7]
 
+
 def git_directory(directory):
 	path = os.path.join(directory, '.git')
 	if os.path.isfile(path):
@@ -28,10 +30,12 @@ def git_directory(directory):
 	else:
 		return path
 
+
 def get_branch_name(base_dir):
 	head = os.path.join(git_directory(base_dir), 'HEAD')
 	return _get_branch_name(base_dir, head, branch_name_from_config_file)
 
+
 def do_status(directory, path, func):
 	if path:
 		gitd = git_directory(directory)
@@ -43,12 +47,14 @@ def do_status(directory, path, func):
 			path, '.gitignore', func, extra_ignore_files=tuple(os.path.join(gitd, x) for x in ('logs/HEAD', 'info/exclude')))
 	return func(directory, path)
 
+
 def ignore_event(path, name):
 	# Ignore changes to the index.lock file, since they happen frequently and
 	# dont indicate an actual change in the working tree status
 	return False
 	return path.endswith('.git') and name == 'index.lock'
 
+
 try:
 	import pygit2 as git
 

powerline/lib/vcs/mercurial.py

@@ -7,6 +7,7 @@ from mercurial import hg, ui, match
 
 from powerline.lib.vcs import get_branch_name, get_file_status
 
+
 def branch_name_from_config_file(directory, config_file):
 	try:
 		with open(config_file, 'rb') as f:
@@ -15,6 +16,7 @@ def branch_name_from_config_file(directory, config_file):
 	except Exception:
 		return 'default'
 
+
 class Repository(object):
 	__slots__ = ('directory', 'ui')
 

powerline/renderer.py

@@ -14,6 +14,7 @@ try:
 except ImportError:
 	pass
 
+
 def construct_returned_value(rendered_highlighted, segments, output_raw):
 	if output_raw:
 		return rendered_highlighted, ''.join((segment['_rendered_raw'] for segment in segments))
@@ -70,7 +71,7 @@ class Renderer(object):
 
 	character_translations = {ord(' '): NBSP}
 	'''Character translations for use in escape() function.
-	
+
 	See documentation of ``unicode.translate`` for details.
 	'''
 
@@ -111,7 +112,7 @@ class Renderer(object):
 
 	def strwidth(self, string):
 		'''Function that returns string width.
-		
+
 		Is used to calculate the place given string occupies when handling 
 		``width`` argument to ``.render()`` method. Must take east asian width 
 		into account.
@@ -125,7 +126,7 @@ class Renderer(object):
 
 	def get_theme(self, matcher_info):
 		'''Get Theme object.
-		
+
 		Is to be overridden by subclasses to support local themes, this variant 
 		only returns ``.theme`` attribute.
 
@@ -152,7 +153,7 @@ class Renderer(object):
 
 	def get_segment_info(self, segment_info, mode):
 		'''Get segment information.
-		
+
 		Must return a dictionary containing at least ``home``, ``environ`` and 
 		``getcwd`` keys (see documentation for ``segment_info`` attribute). This 
 		implementation merges ``segment_info`` dictionary passed to 

powerline/renderers/vim.py

@@ -140,12 +140,12 @@ class VimRenderer(Renderer):
 					hl_group['attr'].append('italic')
 				if attr & ATTR_UNDERLINE:
 					hl_group['attr'].append('underline')
-			hl_group['name'] = 'Pl_' + \
+			hl_group['name'] = ('Pl_' +
-				str(hl_group['ctermfg']) + '_' + \
+						str(hl_group['ctermfg']) + '_' +
-				str(hl_group['guifg']) + '_' + \
+						str(hl_group['guifg']) + '_' +
-				str(hl_group['ctermbg']) + '_' + \
+						str(hl_group['ctermbg']) + '_' +
-				str(hl_group['guibg']) + '_' + \
+						str(hl_group['guibg']) + '_' +
-				''.join(hl_group['attr'])
+						''.join(hl_group['attr']))
 			self.hl_groups[(fg, bg, attr)] = hl_group
 			vim.command('hi {group} ctermfg={ctermfg} guifg={guifg} guibg={guibg} ctermbg={ctermbg} cterm={attr} gui={attr}'.format(
 				group=hl_group['name'],

powerline/segments/common.py

@@ -75,17 +75,22 @@ def branch(pl, segment_info, status_colors=False):
 
 
 @requires_segment_info
-def cwd(pl, segment_info, dir_shorten_len=None, dir_limit_depth=None, use_path_separator=False):
+def cwd(pl, segment_info, dir_shorten_len=None, dir_limit_depth=None, use_path_separator=False, ellipsis='⋯'):
 	'''Return the current working directory.
 
 	Returns a segment list to create a breadcrumb-like effect.
 
 	:param int dir_shorten_len:
-		shorten parent directory names to this length (e.g. :file:`/long/path/to/powerline` → :file:`/l/p/t/powerline`)
+		shorten parent directory names to this length (e.g. 
+		:file:`/long/path/to/powerline` → :file:`/l/p/t/powerline`)
 	:param int dir_limit_depth:
-		limit directory depth to this number (e.g. :file:`/long/path/to/powerline` → :file:`⋯/to/powerline`)
+		limit directory depth to this number (e.g. 
+		:file:`/long/path/to/powerline` → :file:`⋯/to/powerline`)
 	:param bool use_path_separator:
 		Use path separator in place of soft divider.
+	:param str ellipsis:
+		Specifies what to use in place of omitted directories. Use None to not 
+		show this subsegment at all.
 
 	Divider highlight group used: ``cwd:divider``.
 
@@ -110,7 +115,8 @@ def cwd(pl, segment_info, dir_shorten_len=None, dir_limit_depth=None, use_path_s
 	cwd = [i[0:dir_shorten_len] if dir_shorten_len and i else i for i in cwd_split[:-1]] + [cwd_split[-1]]
 	if dir_limit_depth and cwd_split_len > dir_limit_depth + 1:
 		del(cwd[0:-dir_limit_depth])
-		cwd.insert(0, '⋯')
+		if ellipsis is not None:
+			cwd.insert(0, ellipsis)
 	ret = []
 	if not cwd[0]:
 		cwd[0] = '/'
@@ -138,6 +144,8 @@ def date(pl, format='%Y-%m-%d', istime=False):
 
 	:param str format:
 		strftime-style date format string
+	:param bool istime:
+		If true then segment uses ``time`` highlight group.
 
 	Divider highlight group used: ``time:divider``.
 
@@ -150,8 +158,19 @@ def date(pl, format='%Y-%m-%d', istime=False):
 	}]
 
 
-def fuzzy_time(pl):
+UNICODE_TEXT_TRANSLATION = {
-	'''Display the current time as fuzzy time, e.g. "quarter past six".'''
+	ord('\''): '’',
+	ord('-'): '‐',
+}
+
+
+def fuzzy_time(pl, unicode_text=True):
+	'''Display the current time as fuzzy time, e.g. "quarter past six".
+
+	:param bool unicode_text:
+		If true then hyphenminuses (regular ASCII ``-``) and single quotes are 
+		replaced with unicode dashes and apostrophes.
+	'''
 	hour_str = ['twelve', 'one', 'two', 'three', 'four', 'five', 'six', 'seven', 'eight', 'nine', 'ten', 'eleven']
 	minute_str = {
 		5: 'five past',
@@ -194,10 +213,15 @@ def fuzzy_time(pl):
 
 	minute = int(round(now.minute / 5.0) * 5)
 	if minute == 60 or minute == 0:
-		return ' '.join([hour, 'o\'clock'])
+		result = ' '.join([hour, 'o\'clock'])
 	else:
 		minute = minute_str[minute]
-		return ' '.join([minute, hour])
+		result = ' '.join([minute, hour])
+
+	if unicode_text:
+		result = result.translate(UNICODE_TEXT_TRANSLATION)
+
+	return result
 
 
 def _external_ip(query_url='http://ipv4.icanhazip.com/'):
@@ -223,15 +247,15 @@ class ExternalIpSegment(ThreadedSegment):
 external_ip = with_docstring(ExternalIpSegment(),
 '''Return external IP address.
 
-Suggested URIs:
-
-* http://ipv4.icanhazip.com/
-* http://ipv6.icanhazip.com/
-* http://icanhazip.com/ (returns IPv6 address if available, else IPv4)
-
 :param str query_url:
 	URI to query for IP address, should return only the IP address as a text string
 
+	Suggested URIs:
+
+	* http://ipv4.icanhazip.com/
+	* http://ipv6.icanhazip.com/
+	* http://icanhazip.com/ (returns IPv6 address if available, else IPv4)
+
 Divider highlight group used: ``background:divider``.
 ''')
 

powerline/segments/shell.py

@@ -46,7 +46,7 @@ def last_pipe_status(pl, segment_info):
 @requires_segment_info
 def mode(pl, segment_info, override={'vicmd': 'COMMND', 'viins': 'INSERT'}, default=None):
 	'''Return the current mode.
-	
+
 	:param dict override:
 		dict for overriding mode strings.
 	:param str default:

powerline/segments/vim.py

@@ -161,7 +161,6 @@ def file_directory(pl, segment_info, shorten_user=True, shorten_cwd=True, shorte
 	name = buffer_name(segment_info['buffer'])
 	if not name:
 		return None
-	import sys
 	file_directory = vim_funcs['fnamemodify'](name, (':~' if shorten_user else '')
 												+ (':.' if shorten_cwd else '') + ':h')
 	if not file_directory:
@@ -285,7 +284,7 @@ def line_percent(pl, segment_info, gradient=False):
 
 
 @window_cached
-def position(pl, position_strings={'top':'Top', 'bottom':'Bot', 'all':'All'}, gradient=False):
+def position(pl, position_strings={'top': 'Top', 'bottom': 'Bot', 'all': 'All'}, gradient=False):
 	'''Return the position of the current view in the file as a percentage.
 
 	:param dict position_strings:
@@ -368,6 +367,7 @@ def modified_buffers(pl, text='+ ', join_str=','):
 		return text + join_str.join(buffer_mod)
 	return None
 
+
 @requires_segment_info
 def branch(pl, segment_info, status_colors=False):
 	'''Return the current working branch.
@@ -395,6 +395,7 @@ def branch(pl, segment_info, status_colors=False):
 				'divider_highlight_group': 'branch:divider',
 			}]
 
+
 @requires_segment_info
 def file_vcs_status(pl, segment_info):
 	'''Return the VCS status for this buffer.

powerline/theme.py

@@ -1,7 +1,7 @@
 # vim:fileencoding=utf-8:noet
 
 from powerline.segment import gen_segment_getter
-from powerline.lib.unicode import u, unicode
+from powerline.lib.unicode import u
 
 
 def requires_segment_info(func):

tests/test_cmdline.py

@@ -18,9 +18,11 @@ class TestParser(TestCase):
 		parser = get_argparser()
 		out = StrIO()
 		err = StrIO()
+
 		def flush():
 			out.truncate(0)
 			err.truncate(0)
+
 		with replace_attr(sys, 'stdout', out, 'stderr', err):
 			for raising_args, raising_reg in [
 				([],                                     'too few arguments|the following arguments are required: ext'),
@@ -69,7 +71,7 @@ class TestParser(TestCase):
 					'-c', 'common.spaces=4',
 					'-t', 'default.segment_data.hostname.before=H:',
 					'-p', '.',
-					'-R', 'smth={"abc":"def"}'
+					'-R', 'smth={"abc":"def"}',
 				], {
 					'ext': ['shell'],
 					'side': 'left',

tests/test_segments.py

@@ -257,10 +257,24 @@ class TestCommon(TestCase):
 				{'contents': '⋯', 'divider_highlight_group': 'cwd:divider', 'draw_inner_divider': True},
 				{'contents': 'bar', 'divider_highlight_group': 'cwd:divider', 'draw_inner_divider': True, 'highlight_group': ['cwd:current_folder', 'cwd']}
 			])
+			self.assertEqual(common.cwd(pl=pl, segment_info=segment_info, dir_limit_depth=1, ellipsis='...'), [
+				{'contents': '...', 'divider_highlight_group': 'cwd:divider', 'draw_inner_divider': True},
+				{'contents': 'bar', 'divider_highlight_group': 'cwd:divider', 'draw_inner_divider': True, 'highlight_group': ['cwd:current_folder', 'cwd']}
+			])
+			self.assertEqual(common.cwd(pl=pl, segment_info=segment_info, dir_limit_depth=1, ellipsis=None), [
+				{'contents': 'bar', 'divider_highlight_group': 'cwd:divider', 'draw_inner_divider': True, 'highlight_group': ['cwd:current_folder', 'cwd']}
+			])
 			self.assertEqual(common.cwd(pl=pl, segment_info=segment_info, dir_limit_depth=1, use_path_separator=True), [
 				{'contents': '⋯/', 'divider_highlight_group': 'cwd:divider', 'draw_inner_divider': False},
 				{'contents': 'bar', 'divider_highlight_group': 'cwd:divider', 'draw_inner_divider': False, 'highlight_group': ['cwd:current_folder', 'cwd']}
 			])
+			self.assertEqual(common.cwd(pl=pl, segment_info=segment_info, dir_limit_depth=1, use_path_separator=True, ellipsis='...'), [
+				{'contents': '.../', 'divider_highlight_group': 'cwd:divider', 'draw_inner_divider': False},
+				{'contents': 'bar', 'divider_highlight_group': 'cwd:divider', 'draw_inner_divider': False, 'highlight_group': ['cwd:current_folder', 'cwd']}
+			])
+			self.assertEqual(common.cwd(pl=pl, segment_info=segment_info, dir_limit_depth=1, use_path_separator=True, ellipsis=None), [
+				{'contents': 'bar', 'divider_highlight_group': 'cwd:divider', 'draw_inner_divider': False, 'highlight_group': ['cwd:current_folder', 'cwd']}
+			])
 			self.assertEqual(common.cwd(pl=pl, segment_info=segment_info, dir_limit_depth=2, dir_shorten_len=2), [
 				{'contents': '~', 'divider_highlight_group': 'cwd:divider', 'draw_inner_divider': True},
 				{'contents': 'fo', 'divider_highlight_group': 'cwd:divider', 'draw_inner_divider': True},
@@ -312,9 +326,13 @@ class TestCommon(TestCase):
 			time.minute = 59
 			self.assertEqual(common.fuzzy_time(pl=pl), 'round about midnight')
 			time.minute = 33
-			self.assertEqual(common.fuzzy_time(pl=pl), 'twenty-five to twelve')
+			self.assertEqual(common.fuzzy_time(pl=pl), 'twenty‐five to twelve')
 			time.minute = 60
-			self.assertEqual(common.fuzzy_time(pl=pl), 'twelve o\'clock')
+			self.assertEqual(common.fuzzy_time(pl=pl), 'twelve o’clock')
+			time.minute = 33
+			self.assertEqual(common.fuzzy_time(pl=pl, unicode_text=False), 'twenty-five to twelve')
+			time.minute = 60
+			self.assertEqual(common.fuzzy_time(pl=pl, unicode_text=False), 'twelve o\'clock')
 
 	def test_external_ip(self):
 		pl = Pl()
@@ -476,8 +494,8 @@ class TestCommon(TestCase):
 
 	def test_environment(self):
 		pl = Pl()
-		variable = 'FOO';
+		variable = 'FOO'
-		value = 'bar';
+		value = 'bar'
 		with replace_env(variable, value) as segment_info:
 			self.assertEqual(common.environment(pl=pl, segment_info=segment_info, variable=variable), value)
 			segment_info['environ'].pop(variable)
@@ -542,6 +560,7 @@ class TestCommon(TestCase):
 				}
 			])
 
+
 class TestVim(TestCase):
 	def test_mode(self):
 		pl = Pl()
@@ -659,10 +678,10 @@ class TestVim(TestCase):
 			vim_module._set_cursor(0, 0)
 			self.assertEqual(vim.position(pl=pl, segment_info=segment_info), 'Top')
 			vim_module._set_cursor(97, 0)
-			self.assertEqual(vim.position(pl=pl, segment_info=segment_info, position_strings={'top':'Comienzo', 'bottom':'Final', 'all':'Todo'}), 'Final')
+			self.assertEqual(vim.position(pl=pl, segment_info=segment_info, position_strings={'top': 'Comienzo', 'bottom': 'Final', 'all': 'Todo'}), 'Final')
 			segment_info['buffer'][0:-1] = [str(i) for i in range(2)]
 			vim_module._set_cursor(0, 0)
-			self.assertEqual(vim.position(pl=pl, segment_info=segment_info, position_strings={'top':'Comienzo', 'bottom':'Final', 'all':'Todo'}), 'Todo')
+			self.assertEqual(vim.position(pl=pl, segment_info=segment_info, position_strings={'top': 'Comienzo', 'bottom': 'Final', 'all': 'Todo'}), 'Todo')
 			self.assertEqual(vim.position(pl=pl, segment_info=segment_info, gradient=True),
 					[{'contents': 'All', 'highlight_group': ['position_gradient', 'position'], 'gradient_level': 0.0}])
 		finally:

tests/test_shells/postproc.py

@@ -38,14 +38,14 @@ with codecs.open(fname, 'r', encoding='utf-8') as R:
 				try:
 					start = line.index('\033[0;')
 					end = line.index('\033[0m', start)
-					line = line[start : end+4] + '\n'
+					line = line[start:end + 4] + '\n'
 				except ValueError:
 					line = ''
 			elif shell == 'tcsh':
 				try:
 					start = line.index('\033[0;')
 					end = line.index(' ', start)
-					line = line[start : end] + '\033[0m\n'
+					line = line[start:end] + '\033[0m\n'
 				except ValueError:
 					line = ''
 			W.write(line)

tests/vim.py

@@ -307,9 +307,9 @@ def _emul_line(expr):
 	cursorline = windows[_window - 1].cursor[0] + 1
 	numlines = len(_buf_lines[_buffer()])
 	if expr == 'w0':
-		return max(cursorline-5, 1)
+		return max(cursorline - 5, 1)
 	if expr == 'w$':
-		return min(cursorline+5, numlines)
+		return min(cursorline + 5, numlines)
 	raise NotImplementedError
 
 
@@ -401,7 +401,6 @@ class _Buffer(object):
 			import os
 			if type(name) is not bytes:
 				name = name.encode('utf-8')
-			import sys
 			self._name = os.path.abspath(name)
 
 	def __getitem__(self, line):

tools/generate_gradients.py

@@ -138,8 +138,8 @@ palettes = {
 	'256': (cterm_to_hex, lambda c: c),
 	None: (cterm_to_hex[16:], lambda c: c + 16),
 }
-r = [get_rgb(*color) for color in gradient]
+r = [get_rgb(*col) for col in gradient]
-r2 = [find_color(color, *palettes[args.palette])[0] for color in gradient]
+r2 = [find_color(col, *palettes[args.palette])[0] for col in gradient]
 r3 = [i[0] for i in groupby(r2)]
 print(json.dumps(r))
 print(json.dumps(r2))