Initial commit (forked from jamiepine/voicebox)

backend/tests/test_amd_gpu_detect.py

@@ -0,0 +1,96 @@
+"""
+Phase 2.1 Test: AMD GPU detection on Windows.
+
+Validates is_amd_gpu_windows() via mocked WMI and torch queries.
+
+Usage:
+    python -m pytest backend/tests/test_amd_gpu_detect.py -v
+"""
+
+from unittest.mock import MagicMock, patch
+
+import pytest
+
+from backend.utils.platform_detect import is_amd_gpu_windows
+
+
+class TestAmdGpuWindows:
+    """Unit tests for is_amd_gpu_windows with mocks."""
+
+    @pytest.fixture(autouse=True)
+    def _clear_detection_cache(self):
+        # is_amd_gpu_windows is memoized; reset between cases so each mock takes effect.
+        is_amd_gpu_windows.cache_clear()
+        yield
+        is_amd_gpu_windows.cache_clear()
+
+    @patch("backend.utils.platform_detect.platform.system", return_value="Linux")
+    def test_returns_false_on_linux(self, _mock_system):
+        """Non-Windows platforms should always return False."""
+        assert is_amd_gpu_windows() is False
+
+    @patch("backend.utils.platform_detect.platform.system", return_value="Windows")
+    @patch(
+        "backend.utils.platform_detect.subprocess.run",
+        return_value=MagicMock(stdout="1\n", returncode=0),
+    )
+    def test_detects_amd_via_wmi(self, _mock_run, _mock_system):
+        """WMI reporting an AMD adapter should return True."""
+        assert is_amd_gpu_windows() is True
+
+    @patch("backend.utils.platform_detect.platform.system", return_value="Windows")
+    @patch(
+        "backend.utils.platform_detect.subprocess.run",
+        return_value=MagicMock(stdout="0\n", returncode=0),
+    )
+    def test_no_amd_via_wmi(self, _mock_run, _mock_system):
+        """WMI reporting zero AMD adapters should return False."""
+        assert is_amd_gpu_windows() is False
+
+    @patch("backend.utils.platform_detect.platform.system", return_value="Windows")
+    @patch(
+        "backend.utils.platform_detect.subprocess.run",
+        side_effect=Exception("WMI not available"),
+    )
+    @patch("torch.cuda.is_available", return_value=True)
+    @patch(
+        "torch.cuda.get_device_name",
+        return_value="AMD Radeon RX 7800 XT",
+    )
+    def test_fallback_to_torch_radeon(self, _mock_name, _mock_avail, _mock_run, _mock_system):
+        """When WMI fails, torch.cuda.get_device_name('Radeon') should return True."""
+        assert is_amd_gpu_windows() is True
+
+    @patch("backend.utils.platform_detect.platform.system", return_value="Windows")
+    @patch(
+        "backend.utils.platform_detect.subprocess.run",
+        side_effect=Exception("WMI not available"),
+    )
+    @patch("torch.cuda.is_available", return_value=True)
+    @patch(
+        "torch.cuda.get_device_name",
+        return_value="NVIDIA GeForce RTX 4090",
+    )
+    def test_fallback_to_torch_nvidia(self, _mock_name, _mock_avail, _mock_run, _mock_system):
+        """When WMI fails, torch.cuda.get_device_name('NVIDIA') should return False."""
+        assert is_amd_gpu_windows() is False
+
+    @patch("backend.utils.platform_detect.platform.system", return_value="Windows")
+    @patch(
+        "backend.utils.platform_detect.subprocess.run",
+        side_effect=Exception("WMI not available"),
+    )
+    @patch("torch.cuda.is_available", return_value=False)
+    def test_no_torch_cuda(self, _mock_avail, _mock_run, _mock_system):
+        """When WMI fails and torch.cuda is unavailable, should return False."""
+        assert is_amd_gpu_windows() is False
+
+    @patch("backend.utils.platform_detect.platform.system", return_value="Windows")
+    @patch(
+        "backend.utils.platform_detect.subprocess.run",
+        side_effect=Exception("WMI not available"),
+    )
+    def test_torch_not_installed(self, _mock_run, _mock_system):
+        """When torch is not installed, should return False without crashing."""
+        with patch.dict("sys.modules", {"torch": None}):
+            assert is_amd_gpu_windows() is False