pandas-import
Ast Rule: import
pandas-import
function visit(node, filename, code) {
if (node.astType === "importstatement") {
node.packages.forEach(statement => {
if (statement.name && statement.name.value === "pandas") {
if (!statement.as) {
const error = buildError(statement.start.line, statement.start.col, statement.end.line, statement.end.col,
"Must import as pd", "INFO", "BEST_PRACTICE");
const edit = buildEditAdd(statement.end.line, statement.end.col, " as pd");
const fix = buildFix("import as pd", [edit]);
addError(error.addFix(fix));
}
if (statement.as && statement.as.value !== "pd") {
const error = buildError(statement.start.line, statement.start.col, statement.end.line, statement.end.col,
"Must import as pd", "INFO", "BEST_PRACTICE");
const edit = buildEditUpdate(statement.as.start.line, statement.as.start.col,
statement.as.end.line, statement.as.end.col, "pd");
const fix = buildFix(`import as pd`, [edit]);
addError(error.addFix(fix));
}
}
})
}
}
correct.py
Expected test result: no error
error1.py
Expected test result: no error
error2.py
Expected test result: no error