add a bad test case

Author: ntBre

crates/ruff_linter/resources/test/fixtures/pylint/import_private_name/submodule/__main__.py

@@ -44,3 +44,5 @@ class Class:
 
     def __init__(self, arg: vv) -> "zz":
         pass
+
+from foo.    _bar import baz

crates/ruff_linter/src/rules/pylint/rules/import_private_name.rs

@@ -5,7 +5,8 @@ use itertools::Itertools;
 use ruff_macros::{ViolationMetadata, derive_message_formats};
 use ruff_python_ast::{self as ast, helpers::is_dunder, name::QualifiedName};
 use ruff_python_semantic::{FromImport, Import, Imported, ResolvedReference, Scope};
-use ruff_text_size::{Ranged, TextLen, TextRange, TextSize};
+use ruff_python_trivia::{SimpleTokenKind, SimpleTokenizer};
+use ruff_text_size::Ranged;
 
 use crate::Violation;
 use crate::checkers::ast::Checker;
@@ -141,12 +142,15 @@ pub(crate) fn import_private_name(checker: &Checker, scope: &Scope) {
             Some(ast::Stmt::ImportFrom(ast::StmtImportFrom { module, names, .. })) => {
                 if index < import_info.module_name.len() {
                     module.as_ref().map_or(binding.range(), |module| {
-                        let offset: TextSize = import_info.qualified_name.segments()[..index]
-                            .iter()
-                            .map(|segment| segment.text_len() + TextSize::new(1))
-                            .sum();
-
-                        TextRange::at(module.start() + offset, private_name.text_len())
+                        SimpleTokenizer::starts_at(module.start(), checker.locator().contents())
+                            .skip_trivia()
+                            .find(|token| {
+                                token.kind == SimpleTokenKind::Name
+                                    && checker.locator().slice(token.range()) == *private_name
+                            })
+                            .map_or(binding.range(), |token: ruff_python_trivia::SimpleToken| {
+                                token.range()
+                            })
                     })
                 } else {
                     names

crates/ruff_linter/src/rules/pylint/snapshots/ruff_linter__rules__pylint__tests__PLC2701_import_private_name__submodule____main__.py.snap

@@ -75,3 +75,12 @@ PLC2701 Private name import `_ddd` from external module `bbb.ccc`
  9 |
 10 | # Non-errors.
    |
+
+PLC2701 Private name import `_bar` from external module `foo`
+  --> __main__.py:48:14
+   |
+46 |         pass
+47 |
+48 | from foo.    _bar import baz
+   |              ^^^^
+   |