wehub-resource-sync
39 lines
1.0 KiB
Python
39 lines
1.0 KiB
Python
import unittest
|
|
|
|
import yaml
|
|
|
|
from ray.autoscaler._private.providers import (
|
|
_DEFAULT_CONFIGS,
|
|
_NODE_PROVIDERS,
|
|
_PROVIDER_PRETTY_NAMES,
|
|
)
|
|
|
|
|
|
class TestProviders(unittest.TestCase):
|
|
def test_node_providers(self):
|
|
for provider_name, provider_cls in _NODE_PROVIDERS.items():
|
|
config = {"module": "ray.autoscaler._private"}
|
|
|
|
try:
|
|
provider_cls(config)
|
|
except ImportError as e:
|
|
if f"ray.autoscaler.{provider_name}" in str(e):
|
|
self.fail(
|
|
f"Unexpected import error for provider {provider_name}: {e}"
|
|
)
|
|
|
|
def test_provider_pretty_names(self):
|
|
self.assertEqual(
|
|
set(_NODE_PROVIDERS.keys()), set(_PROVIDER_PRETTY_NAMES.keys())
|
|
)
|
|
|
|
def test_default_configs(self):
|
|
for config_loader in _DEFAULT_CONFIGS.values():
|
|
config_path = config_loader()
|
|
with open(config_path) as f:
|
|
yaml.safe_load(f)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|