Fix failing test

Author: AlexWaygood

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

@@ -244,7 +244,7 @@ mod tests {
     fn f401_preview_first_party_submodule(contents: &str, snapshot: &str) {
         let diagnostics = test_contents(
             &SourceKind::Python(dedent(contents).to_string()),
-            Path::new("__init__.py"),
+            Path::new("f401_preview_first_party_submodule/__init__.py"),
             &LinterSettings {
                 preview: PreviewMode::Enabled,
                 isort: isort::settings::Settings {

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

@@ -305,7 +305,7 @@ pub(crate) fn unused_import(checker: &Checker, scope: &Scope, diagnostics: &mut
             .map(|binding| {
                 let context = if in_except_handler {
                     Some(UnusedImportContext::ExceptHandler)
-                } else if in_init {
+                } else if in_init && !binding.import.is_submodule_import() {
                     Some(UnusedImportContext::Init {
                         first_party: is_first_party(
                             &binding.import.qualified_name().to_string(),

crates/ruff_linter/src/rules/pyflakes/snapshots/ruff_linter__rules__pyflakes__tests__f401_preview_first_party_submodule.snap

@@ -0,0 +1,17 @@
+---
+source: crates/ruff_linter/src/rules/pyflakes/mod.rs
+---
+__init__.py:2:8: F401 [*] `submodule.a` imported but unused
+  |
+2 | import submodule.a
+  |        ^^^^^^^^^^^ F401
+3 | __all__ = ['FOO']
+4 | FOO = 42
+  |
+  = help: Remove unused import: `submodule.a`
+
+ℹ Unsafe fix
+1 1 | 
+2   |-import submodule.a
+3 2 | __all__ = ['FOO']
+4 3 | FOO = 42