tests/test_config.py

import sys

def test_imports():
    try:
        import ultralytics
        import cv2
        import PIL
        import folium
        print(f"Ultralytics installed successfully: {ultralytics.__version__}")
        assert ultralytics.__version__ == '8.2.18', "YOLO Version mismatch"
        print("Initialization successful.")
    except Exception as e:
        print(f"Error during initialization: {e}")
        sys.exit(1)

if __name__ == "__main__":
    test_imports()
    print("ALL TESTS PASSED")