def test_correct_slots( assert_errors, parse_ast_tree, default_options, code, ): """Testing that correct slots are allowed.""" tree = parse_ast_tree(class_body_template.format(code)) visitor = WrongSlotsVisitor(default_options, tree=tree) visitor.run() assert_errors(visitor, [])
def test_incorrect_slots( assert_errors, parse_ast_tree, default_options, code, template, ): """Testing that incorrect slots are prohibited.""" tree = parse_ast_tree(template.format(code)) visitor = WrongSlotsVisitor(default_options, tree=tree) visitor.run() assert_errors(visitor, [WrongSlotsViolation])