Add test

Author: AlexWaygood

crates/ruff_linter/resources/test/fixtures/pyflakes/F401_30.py

@@ -0,0 +1,6 @@
+"""
+Test: ensure we're able to correctly remove unused imports
+even if they have characters in them that undergo NFKC normalization
+"""
+
+from .main import MaµToMan

crates/ruff_linter/src/fix/codemods.rs

@@ -170,33 +170,35 @@ pub(crate) fn retain_imports(
     Ok(tree.codegen_stylist(stylist))
 }
 
-fn collect_segments<'a>(expr: &'a Expression, parts: &mut SmallVec<[&'a str; 8]>) {
-    match expr {
-        Expression::Call(expr) => {
-            collect_segments(&expr.func, parts);
-        }
-        Expression::Attribute(expr) => {
-            collect_segments(&expr.value, parts);
-            parts.push(expr.attr.value);
-        }
-        Expression::Name(expr) => {
-            parts.push(expr.value);
+fn qualified_name_from_name_or_attribute(module: &NameOrAttribute) -> String {
+    fn collect_segments<'a>(expr: &'a Expression, parts: &mut SmallVec<[&'a str; 8]>) {
+        match expr {
+            Expression::Call(expr) => {
+                collect_segments(&expr.func, parts);
+            }
+            Expression::Attribute(expr) => {
+                collect_segments(&expr.value, parts);
+                parts.push(expr.attr.value);
+            }
+            Expression::Name(expr) => {
+                parts.push(expr.value);
+            }
+            _ => {}
         }
-        _ => {}
     }
-}
 
-fn unqualified_name_from_expression<'a>(expr: &'a Expression<'a>) -> Option<UnqualifiedName<'a>> {
-    let mut segments = smallvec![];
-    collect_segments(expr, &mut segments);
-    if segments.is_empty() {
-        None
-    } else {
-        Some(segments.into_iter().collect())
+    fn unqualified_name_from_expression<'a>(
+        expr: &'a Expression<'a>,
+    ) -> Option<UnqualifiedName<'a>> {
+        let mut segments = smallvec![];
+        collect_segments(expr, &mut segments);
+        if segments.is_empty() {
+            None
+        } else {
+            Some(segments.into_iter().collect())
+        }
     }
-}
 
-fn qualified_name_from_name_or_attribute(module: &NameOrAttribute) -> String {
     let unnormalized = match module {
         NameOrAttribute::N(name) => Cow::Borrowed(name.value),
         NameOrAttribute::A(attr) => {
@@ -208,5 +210,6 @@ fn qualified_name_from_name_or_attribute(module: &NameOrAttribute) -> String {
             )
         }
     };
+
     unnormalized.nfkc().collect()
 }

crates/ruff_linter/src/rules/pyflakes/mod.rs

@@ -258,6 +258,7 @@ mod tests {
     #[test_case(Rule::UnusedImport, Path::new("F401_27__all_mistyped/__init__.py"))]
     #[test_case(Rule::UnusedImport, Path::new("F401_28__all_multiple/__init__.py"))]
     #[test_case(Rule::UnusedImport, Path::new("F401_29__all_conditional/__init__.py"))]
+    #[test_case(Rule::UnusedImport, Path::new("F401_30.py"))]
     fn f401_deprecated_option(rule_code: Rule, path: &Path) -> Result<()> {
         let snapshot = format!(
             "{}_deprecated_option_{}",

crates/ruff_linter/src/rules/pyflakes/snapshots/ruff_linter__rules__pyflakes__tests__F401_deprecated_option_F401_30.py.snap

@@ -0,0 +1,17 @@
+---
+source: crates/ruff_linter/src/rules/pyflakes/mod.rs
+---
+F401_30.py:6:19: F401 [*] `.main.MaμToMan` imported but unused
+  |
+4 | """
+5 | 
+6 | from .main import MaµToMan
+  |                   ^^^^^^^^ F401
+  |
+  = help: Remove unused import: `.main.MaμToMan`
+
+ℹ Safe fix
+3 3 | even if they have characters in them that undergo NFKC normalization
+4 4 | """
+5 5 | 
+6   |-from .main import MaµToMan