def test_wrong_return_annotation(
assert_errors,
parse_ast_tree,
code,
default_options,
mode,
):
"""Ensures that using incorrect return annotations is forbiden."""
tree = parse_ast_tree(mode(code))
visitor = WrongAnnotationVisitor(default_options, tree=tree)
visitor.run()
assert_errors(visitor, [MultilineFunctionAnnotationViolation])
def test_correct_return_annotation(
assert_errors,
parse_ast_tree,
code,
default_options,
mode,
):
"""Ensures that it is possible to use correct type annotations."""
tree = parse_ast_tree(mode(code))
visitor = WrongAnnotationVisitor(default_options, tree=tree)
visitor.run()
assert_errors(visitor, [])
def test_forbidden_literal_none_annotation(
assert_errors,
parse_ast_tree,
code,
default_options,
mode,
):
"""Ensures that using incorrect argument annotations is forbiden."""
tree = parse_ast_tree(mode(code))
visitor = WrongAnnotationVisitor(default_options, tree=tree)
visitor.run()
assert_errors(visitor, [LiteralNoneViolation])