import sys from transformers.testing_utils import run_test_using_subprocess from transformers.utils.import_utils import clear_import_cache @run_test_using_subprocess def test_clear_import_cache(): """Test the clear_import_cache function.""" # Save initial state initial_modules = {name: mod for name, mod in sys.modules.items() if name.startswith("transformers.")} assert len(initial_modules) > 0, "No transformers modules loaded before test" # Execute clear_import_cache() function clear_import_cache() # Verify modules were removed remaining_modules = {name: mod for name, mod in sys.modules.items() if name.startswith("transformers.")} assert len(remaining_modules) < len(initial_modules), "No modules were removed" # Import and verify module exists from transformers.models.auto import modeling_auto assert "transformers.models.auto.modeling_auto" in sys.modules assert modeling_auto.__name__ == "transformers.models.auto.modeling_auto"