You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
20 lines
710 B
20 lines
710 B
import pytest
|
|
|
|
|
|
def test_multi_model_gpu_manager_module_can_be_imported():
|
|
"""Test that multi_model_gpu_manager module can be imported"""
|
|
try:
|
|
from swarms.structs import multi_model_gpu_manager
|
|
assert multi_model_gpu_manager is not None
|
|
except ImportError:
|
|
pytest.skip("Module has import dependencies or GPU requirements")
|
|
|
|
|
|
def test_multi_model_gpu_manager_class_exists():
|
|
"""Test that MultiModelGPUManager class exists"""
|
|
try:
|
|
from swarms.structs.multi_model_gpu_manager import MultiModelGPUManager
|
|
assert MultiModelGPUManager is not None
|
|
except (ImportError, AttributeError):
|
|
pytest.skip("Class not available or has GPU dependencies")
|