mirror of
https://github.com/huggingface/transformers.git
synced 2025-07-03 12:50:06 +06:00
8e67230860
* test fixup * test fixup * fixing tests for unused imports * style fixes * fix * style fixes * styke fix * remove isolated module cache * rm custom subprocess defination * run using exsiting fn * style fixup * make fixup * remove redundant comments * rm redundat skipif + style changes
27 lines
1001 B
Python
27 lines
1001 B
Python
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"
|