mirror of
https://fuchsia.googlesource.com/third_party/github.com/pylint-dev/pylint
synced 2024-09-22 00:24:37 +00:00
211d81f84f
This was changed automatically in #894, but apparently we need to have the copyright notice somewhere.
251 lines
9.1 KiB
Python
251 lines
9.1 KiB
Python
# Copyright (c) 2003-2016 LOGILAB S.A. (Paris, FRANCE).
|
|
# http://www.logilab.fr/ -- mailto:contact@logilab.fr
|
|
# Licensed under the GPL: https://www.gnu.org/licenses/old-licenses/gpl-2.0.html
|
|
# For details: https://github.com/PyCQA/pylint/blob/master/COPYING
|
|
|
|
"""Check format checker helper functions"""
|
|
|
|
from __future__ import unicode_literals
|
|
|
|
import astroid
|
|
|
|
from pylint.checkers.format import *
|
|
|
|
from pylint.testutils import (
|
|
CheckerTestCase, Message, set_config, tokenize_str,
|
|
)
|
|
|
|
|
|
class MultiStatementLineTest(CheckerTestCase):
|
|
CHECKER_CLASS = FormatChecker
|
|
|
|
def testSingleLineIfStmts(self):
|
|
stmt = astroid.extract_node("""
|
|
if True: pass #@
|
|
""")
|
|
with self.assertAddsMessages(Message('multiple-statements', node=stmt.body[0])):
|
|
self.checker.process_tokens([])
|
|
self.checker.visit_default(stmt.body[0])
|
|
self.checker.config.single_line_if_stmt = True
|
|
with self.assertNoMessages():
|
|
self.checker.process_tokens([])
|
|
self.checker.visit_default(stmt.body[0])
|
|
stmt = astroid.extract_node("""
|
|
if True: pass #@
|
|
else:
|
|
pass
|
|
""")
|
|
with self.assertAddsMessages(Message('multiple-statements', node=stmt.body[0])):
|
|
self.checker.process_tokens([])
|
|
self.checker.visit_default(stmt.body[0])
|
|
|
|
def testTryExceptFinallyNoMultipleStatement(self):
|
|
tree = astroid.extract_node("""
|
|
try: #@
|
|
pass
|
|
except:
|
|
pass
|
|
finally:
|
|
pass""")
|
|
with self.assertNoMessages():
|
|
self.checker.process_tokens([])
|
|
self.checker.visit_default(tree.body[0])
|
|
|
|
|
|
|
|
class SuperfluousParenthesesTest(CheckerTestCase):
|
|
CHECKER_CLASS = FormatChecker
|
|
|
|
def testCheckKeywordParensHandlesValidCases(self):
|
|
self.checker._keywords_with_parens = set()
|
|
cases = [
|
|
'if foo:',
|
|
'if foo():',
|
|
'if (x and y) or z:',
|
|
'assert foo()',
|
|
'assert ()',
|
|
'if (1, 2) in (3, 4):',
|
|
'if (a or b) in c:',
|
|
'return (x for x in x)',
|
|
'if (x for x in x):',
|
|
'for x in (x for x in x):',
|
|
'not (foo or bar)',
|
|
'not (foo or bar) and baz',
|
|
]
|
|
with self.assertNoMessages():
|
|
for code in cases:
|
|
self.checker._check_keyword_parentheses(tokenize_str(code), 0)
|
|
|
|
def testCheckKeywordParensHandlesUnnecessaryParens(self):
|
|
self.checker._keywords_with_parens = set()
|
|
cases = [
|
|
(Message('superfluous-parens', line=1, args='if'),
|
|
'if (foo):', 0),
|
|
(Message('superfluous-parens', line=1, args='if'),
|
|
'if ((foo, bar)):', 0),
|
|
(Message('superfluous-parens', line=1, args='if'),
|
|
'if (foo(bar)):', 0),
|
|
(Message('superfluous-parens', line=1, args='return'),
|
|
'return ((x for x in x))', 0),
|
|
(Message('superfluous-parens', line=1, args='not'),
|
|
'not (foo)', 0),
|
|
(Message('superfluous-parens', line=1, args='not'),
|
|
'if not (foo):', 1),
|
|
(Message('superfluous-parens', line=1, args='if'),
|
|
'if (not (foo)):', 0),
|
|
(Message('superfluous-parens', line=1, args='not'),
|
|
'if (not (foo)):', 2),
|
|
]
|
|
for msg, code, offset in cases:
|
|
with self.assertAddsMessages(msg):
|
|
self.checker._check_keyword_parentheses(tokenize_str(code), offset)
|
|
|
|
def testFuturePrintStatementWithoutParensWarning(self):
|
|
code = """from __future__ import print_function
|
|
print('Hello world!')
|
|
"""
|
|
tree = astroid.parse(code)
|
|
with self.assertNoMessages():
|
|
self.checker.process_module(tree)
|
|
self.checker.process_tokens(tokenize_str(code))
|
|
|
|
|
|
class CheckSpaceTest(CheckerTestCase):
|
|
CHECKER_CLASS = FormatChecker
|
|
|
|
def testParenthesesGood(self):
|
|
good_cases = [
|
|
'(a)\n',
|
|
'(a * (b + c))\n',
|
|
'(#\n a)\n',
|
|
]
|
|
with self.assertNoMessages():
|
|
for code in good_cases:
|
|
self.checker.process_tokens(tokenize_str(code))
|
|
|
|
def testParenthesesBad(self):
|
|
with self.assertAddsMessages(
|
|
Message('bad-whitespace', line=1,
|
|
args=('No', 'allowed', 'after', 'bracket', '( a)\n^'))):
|
|
self.checker.process_tokens(tokenize_str('( a)\n'))
|
|
|
|
with self.assertAddsMessages(
|
|
Message('bad-whitespace', line=1,
|
|
args=('No', 'allowed', 'before', 'bracket', '(a )\n ^'))):
|
|
self.checker.process_tokens(tokenize_str('(a )\n'))
|
|
|
|
with self.assertAddsMessages(
|
|
Message('bad-whitespace', line=1,
|
|
args=('No', 'allowed', 'before', 'bracket', 'foo (a)\n ^'))):
|
|
self.checker.process_tokens(tokenize_str('foo (a)\n'))
|
|
|
|
with self.assertAddsMessages(
|
|
Message('bad-whitespace', line=1,
|
|
args=('No', 'allowed', 'before', 'bracket', '{1: 2} [1]\n ^'))):
|
|
self.checker.process_tokens(tokenize_str('{1: 2} [1]\n'))
|
|
|
|
def testTrailingCommaGood(self):
|
|
with self.assertNoMessages():
|
|
self.checker.process_tokens(tokenize_str('(a, )\n'))
|
|
self.checker.process_tokens(tokenize_str('(a,)\n'))
|
|
|
|
self.checker.config.no_space_check = []
|
|
with self.assertNoMessages():
|
|
self.checker.process_tokens(tokenize_str('(a,)\n'))
|
|
|
|
@set_config(no_space_check=[])
|
|
def testTrailingCommaBad(self):
|
|
with self.assertAddsMessages(
|
|
Message('bad-whitespace', line=1,
|
|
args=('No', 'allowed', 'before', 'bracket', '(a, )\n ^'))):
|
|
self.checker.process_tokens(tokenize_str('(a, )\n'))
|
|
|
|
def testComma(self):
|
|
with self.assertAddsMessages(
|
|
Message('bad-whitespace', line=1,
|
|
args=('No', 'allowed', 'before', 'comma', '(a , b)\n ^'))):
|
|
self.checker.process_tokens(tokenize_str('(a , b)\n'))
|
|
|
|
def testSpacesAllowedInsideSlices(self):
|
|
good_cases = [
|
|
'[a:b]\n',
|
|
'[a : b]\n',
|
|
'[a : ]\n',
|
|
'[:a]\n',
|
|
'[:]\n',
|
|
'[::]\n',
|
|
]
|
|
with self.assertNoMessages():
|
|
for code in good_cases:
|
|
self.checker.process_tokens(tokenize_str(code))
|
|
|
|
def testKeywordSpacingGood(self):
|
|
with self.assertNoMessages():
|
|
self.checker.process_tokens(tokenize_str('foo(foo=bar)\n'))
|
|
self.checker.process_tokens(tokenize_str('lambda x=1: x\n'))
|
|
|
|
def testKeywordSpacingBad(self):
|
|
with self.assertAddsMessages(
|
|
Message('bad-whitespace', line=1,
|
|
args=('No', 'allowed', 'before', 'keyword argument assignment',
|
|
'(foo =bar)\n ^'))):
|
|
self.checker.process_tokens(tokenize_str('(foo =bar)\n'))
|
|
|
|
with self.assertAddsMessages(
|
|
Message('bad-whitespace', line=1,
|
|
args=('No', 'allowed', 'after', 'keyword argument assignment',
|
|
'(foo= bar)\n ^'))):
|
|
self.checker.process_tokens(tokenize_str('(foo= bar)\n'))
|
|
|
|
with self.assertAddsMessages(
|
|
Message('bad-whitespace', line=1,
|
|
args=('No', 'allowed', 'around', 'keyword argument assignment',
|
|
'(foo = bar)\n ^'))):
|
|
self.checker.process_tokens(tokenize_str('(foo = bar)\n'))
|
|
|
|
def testOperatorSpacingGood(self):
|
|
good_cases = [
|
|
'a = b\n'
|
|
'a < b\n'
|
|
'a\n< b\n',
|
|
]
|
|
with self.assertNoMessages():
|
|
for code in good_cases:
|
|
self.checker.process_tokens(tokenize_str(code))
|
|
|
|
def testOperatorSpacingBad(self):
|
|
with self.assertAddsMessages(
|
|
Message('bad-whitespace', line=1,
|
|
args=('Exactly one', 'required', 'before', 'comparison', 'a< b\n ^'))):
|
|
self.checker.process_tokens(tokenize_str('a< b\n'))
|
|
|
|
with self.assertAddsMessages(
|
|
Message('bad-whitespace', line=1,
|
|
args=('Exactly one', 'required', 'after', 'comparison', 'a <b\n ^'))):
|
|
self.checker.process_tokens(tokenize_str('a <b\n'))
|
|
|
|
with self.assertAddsMessages(
|
|
Message('bad-whitespace', line=1,
|
|
args=('Exactly one', 'required', 'around', 'comparison', 'a<b\n ^'))):
|
|
self.checker.process_tokens(tokenize_str('a<b\n'))
|
|
|
|
with self.assertAddsMessages(
|
|
Message('bad-whitespace', line=1,
|
|
args=('Exactly one', 'required', 'around', 'comparison', 'a< b\n ^'))):
|
|
self.checker.process_tokens(tokenize_str('a< b\n'))
|
|
|
|
def testEmptyLines(self):
|
|
self.checker.config.no_space_check = []
|
|
with self.assertAddsMessages(
|
|
Message('trailing-whitespace', line=2)):
|
|
self.checker.process_tokens(tokenize_str('a = 1\n \nb = 2\n'))
|
|
|
|
self.checker.config.no_space_check = ['empty-line']
|
|
with self.assertNoMessages():
|
|
self.checker.process_tokens(tokenize_str('a = 1\n \nb = 2\n'))
|
|
|
|
|
|
if __name__ == '__main__':
|
|
import unittest
|
|
unittest.main()
|