def test_multiple_assignments(
assert_errors,
parse_ast_tree,
code,
assignment,
default_options,
):
"""Testing that multiple assignments are restricted."""
tree = parse_ast_tree(code.format(assignment))
visitor = WrongAssignmentVisitor(default_options, tree=tree)
visitor.run()
assert_errors(visitor, [IncorrectUnpackingViolation])
def test_unpacking_to_list(
assert_errors,
parse_ast_tree,
default_options,
code,
assignment,
):
"""Ensure that unpacking iterable to list is restricted."""
tree = parse_ast_tree(code.format(assignment))
visitor = WrongAssignmentVisitor(default_options, tree=tree)
visitor.run()
assert_errors(visitor, [UnpackingIterableToListViolation])
def test_element_getting_by_unpacking(
assert_errors,
parse_ast_tree,
code,
definition,
default_options,
):
"""Testing that getting element by unpacking is restricted."""
tree = parse_ast_tree(code.format(definition))
visitor = WrongAssignmentVisitor(default_options, tree=tree)
visitor.run()
assert_errors(
visitor,
[GettingElementByUnpackingViolation],
ignored_types=WrongUnpackingViolation,
)
