mirror of
https://github.com/huggingface/transformers.git
synced 2025-07-04 05:10:06 +06:00
24 lines
679 B
Python
24 lines
679 B
Python
import sys
|
|
|
|
from transformers.utils.import_utils import clear_import_cache
|
|
|
|
|
|
def test_clear_import_cache():
|
|
# Import some transformers modules
|
|
|
|
# Get initial module count
|
|
initial_modules = {name: mod for name, mod in sys.modules.items() if name.startswith("transformers.")}
|
|
|
|
# Verify we have some modules loaded
|
|
assert len(initial_modules) > 0
|
|
|
|
# Clear cache
|
|
clear_import_cache()
|
|
|
|
# Check 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)
|
|
|
|
# Verify we can reimport
|
|
assert "transformers" in sys.modules
|