[ruff F401 #10390 #10391] fixture tests

crates/ruff_linter/resources/test/fixtures/pyflakes/F401_24/__init__.py

@@ -0,0 +1,36 @@
+'''__init__.py without __all__
+
+Stdlib and third party imports are unsafe removals
+
+First party imports get changed to redundant aliases
+'''
+
+
+
+import os # Ok: is used
+_ = os
+
+
+
+import sys # F401: remove unused
+
+
+
+import used # Ok: is used
+_ = used
+
+
+
+
+
+
+
+import aliased as aliased # Ok: is redundant alias
+
+
+
+import renamed as bees # Ok: is aliased
+
+
+
+import unused # F401: change to redundant alias

crates/ruff_linter/resources/test/fixtures/pyflakes/F401_25_dunder_all/__init__.py

@@ -0,0 +1,40 @@
+'''__init__.py with __all__
+
+First party imports get added to __all__
+
+Stdlib and third party imports are unsafe removals
+'''
+
+
+
+import os # Ok: is used
+_ = os
+
+
+
+import sys # F401: remove unused
+
+
+
+import argparse # Ok: is exported
+
+
+
+import used # Ok: is used
+_ = used
+
+
+
+import aliased as aliased # Ok: is redundant alias
+
+
+
+import renamed as bees # Ok: is aliased
+
+
+
+import unused # F401: add to __all__
+
+
+
+__all__ = ["argparse"]

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

@@ -56,6 +56,8 @@ mod tests {
     #[test_case(Rule::UnusedImport, Path::new("F401_21.py"))]
     #[test_case(Rule::UnusedImport, Path::new("F401_22.py"))]
     #[test_case(Rule::UnusedImport, Path::new("F401_23.py"))]
+    #[test_case(Rule::UnusedImport, Path::new("F401_24/__init__.py"))]
+    #[test_case(Rule::UnusedImport, Path::new("F401_25_dunder_all/__init__.py"))]
     #[test_case(Rule::ImportShadowedByLoopVar, Path::new("F402.py"))]
     #[test_case(Rule::ImportShadowedByLoopVar, Path::new("F402.ipynb"))]
     #[test_case(Rule::UndefinedLocalWithImportStar, Path::new("F403.py"))]

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

@@ -116,7 +116,7 @@ impl Violation for UnusedImport {
 
 fn is_first_party(checker: &Checker, qualified_name: &str) -> bool {
     use isort::{ImportSection, ImportType};
-    match isort::categorize(
+    let category = isort::categorize(
         qualified_name,
         None,
         &checker.settings.src,
@@ -127,7 +127,8 @@ fn is_first_party(checker: &Checker, qualified_name: &str) -> bool {
         checker.settings.isort.no_sections,
         &checker.settings.isort.section_order,
         &checker.settings.isort.default_section,
-    ) {
+    );
+    match category {
         ImportSection::Known(ImportType::FirstParty | ImportType::LocalFolder) => true,
         _ => false,
     }