gh-98254: Include stdlib module names in error messages for NameErrors (#98255)

This commit is contained in:
Pablo Galindo Salgado 2022-10-15 23:13:33 +01:00 committed by GitHub
parent 3a639bbeac
commit bb56dead33
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
6 changed files with 101 additions and 31 deletions

View file

@ -3185,6 +3185,21 @@ class SuggestionFormattingTestBase:
actual = self.get_suggestion(func)
self.assertNotIn("something", actual)
def test_name_error_for_stdlib_modules(self):
def func():
stream = io.StringIO()
actual = self.get_suggestion(func)
self.assertIn("forget to import 'io'", actual)
def test_name_error_for_private_stdlib_modules(self):
def func():
stream = _io.StringIO()
actual = self.get_suggestion(func)
self.assertIn("forget to import '_io'", actual)
class PurePythonSuggestionFormattingTests(
PurePythonExceptionFormattingMixin,