Method test_rocm_initialization

tests/test_gpu.py:209–213

(self, mock_subprocess, mock_nvidia, mock_rocm)

    @mock.patch("codecarbon.core.gpu.is_nvidia_system", return_value=False)
    @mock.patch("codecarbon.core.gpu_amd.subprocess.check_output")
    def test_rocm_initialization(self, mock_subprocess, mock_nvidia, mock_rocm):
        from codecarbon.core.gpu import AllGPUDevices

        # Should not crash on init
        AllGPUDevices()

    @mock.patch("codecarbon.core.gpu.is_rocm_system", return_value=False)
    @mock.patch("codecarbon.core.gpu.is_nvidia_system", return_value=True)