tests/test_devices.py

from arena.core.devices import available_devices, resolve_device


def test_cpu_is_available():
    devices = available_devices()
    assert "cpu" in devices
    assert str(resolve_device("cpu")) == "cpu"