wehub-resource-sync
23 lines
592 B
Python
23 lines
592 B
Python
import sys
|
|
|
|
import pytest
|
|
|
|
from ray.util import accelerators
|
|
from ray.util.annotations import RayDeprecationWarning
|
|
|
|
|
|
def test_accelerators():
|
|
assert accelerators.NVIDIA_TESLA_K80 == "K80"
|
|
assert accelerators.NVIDIA_A100 == "A100"
|
|
with pytest.raises(
|
|
AttributeError,
|
|
match="module 'ray.util.accelerators' has no attribute 'NVIDIA_INVALID'",
|
|
):
|
|
_ = accelerators.NVIDIA_INVALID
|
|
with pytest.warns(RayDeprecationWarning):
|
|
assert accelerators.NVIDIA_TESLA_A100 == "A100"
|
|
|
|
|
|
if __name__ == "__main__":
|
|
sys.exit(pytest.main(["-sv", __file__]))
|