Fix "lazy from (...) import (...)" tests (#145213)

This commit is contained in:
Filipe Laíns 2026-02-25 15:57:06 +00:00 committed by GitHub
parent 017ccd3bf4
commit 80b2b88338
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
3 changed files with 11 additions and 10 deletions

View file

@ -1,11 +1,11 @@
import importlib
import sys
def filter(module_name, imported_name, from_list):
assert module_name == __name__
assert imported_name == "test.test_import.data.lazy_imports.basic2"
assert from_list == ['f']
assert from_list == ('f',)
return False
importlib.set_lazy_imports(None, filter)
sys.set_lazy_imports_filter(filter)
lazy from import test.test_import.data.lazy_imports.basic2 import f
lazy from test.test_import.data.lazy_imports.basic2 import f

View file

@ -1,11 +1,12 @@
import importlib
import sys
def filter(module_name, imported_name, from_list):
assert module_name == __name__
assert imported_name == "test.test_import.data.lazy_imports.basic2"
assert from_list == ['f']
assert from_list == ('f',)
return True
importlib.set_lazy_imports(None, filter)
sys.set_lazy_imports("normal")
sys.set_lazy_imports_filter(filter)
lazy from import test.test_import.data.lazy_imports.basic2 import f
lazy from test.test_import.data.lazy_imports.basic2 import f

View file

@ -119,12 +119,12 @@ def test_global_filter_true(self):
def test_global_filter_from(self):
"""Filter should work with 'from' imports."""
import test.test_import.data.lazy_imports.global_filter
import test.test_import.data.lazy_imports.global_filter_from
self.assertIn("test.test_import.data.lazy_imports.basic2", sys.modules)
def test_global_filter_from_true(self):
"""Filter returning True should allow lazy 'from' imports."""
import test.test_import.data.lazy_imports.global_filter_true
import test.test_import.data.lazy_imports.global_filter_from_true
self.assertNotIn("test.test_import.data.lazy_imports.basic2", sys.modules)