Method test_nvidia_initialization

tests/test_gpu.py:218–222

(self, mock_subprocess, mock_nvidia, mock_rocm)

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

        # Should not crash on init
        AllGPUDevices()