diff --git a/scripts/add_rule.py b/scripts/add_rule.py index 3e4c105822d55..ad478666a437f 100755 --- a/scripts/add_rule.py +++ b/scripts/add_rule.py @@ -20,11 +20,12 @@ def main(*, name: str, prefix: str, code: str, linter: str) -> None: """Generate boilerplate for a new rule.""" # Create a test fixture. + filestem = f"{prefix}{code}" if linter != "pylint" else snake_case(name) with ( ROOT_DIR / "crates/ruff/resources/test/fixtures" / dir_name(linter) - / f"{prefix}{code}.py" + / f"{filestem}.py" ).open( "a", ): @@ -45,7 +46,6 @@ def main(*, name: str, prefix: str, code: str, linter: str) -> None: line.strip() == "fn rules(rule_code: Rule, path: &Path) -> Result<()> {" ): indent = get_indent(line) - filestem = f"{prefix}{code}" if linter != "pylint" else snake_case(name) lines.append( f'{indent}#[test_case(Rule::{name}, Path::new("{filestem}.py"))]', )