File size: 1,794 Bytes
e09a76d |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 |
import os
import importlib.util
import sys
def test_imports(directories):
"""
Test importing all Python files in the specified directories.
Parameters:
- directories: List of directory paths to test.
"""
print("Testing Conda environment...")
for directory in directories:
print(f"\nChecking directory: {directory}")
# Check if the directory exists
if not os.path.isdir(directory):
print(f"Directory not found: {directory}")
continue
# Iterate through all files in the directory
for filename in os.listdir(directory):
# Only consider Python files
if filename.endswith(".py"):
filepath = os.path.join(directory, filename)
module_name = os.path.splitext(filename)[0] # Remove .py extension
try:
# Dynamically import the module
spec = importlib.util.spec_from_file_location(module_name, filepath)
module = importlib.util.module_from_spec(spec)
spec.loader.exec_module(module)
print(f"Successfully imported: {filepath}")
except Exception as e:
# Print the file and the error message if import fails
print(f"Failed to import: {filepath}")
print(f"Error: {e}")
if __name__ == "__main__":
# Automatically append the current directory to sys.path
current_directory = os.getcwd()
sys.path.append(current_directory)
print(f"Current directory added to sys.path: {current_directory}")
# List of directories to check
directories = ["scripts", "root_gnn_base", "models"]
test_imports(directories) |