wehub-resource-sync
131 lines
4.1 KiB
Python
131 lines
4.1 KiB
Python
"""Tests for Linux Intel integrated GPU detection."""
|
|
|
|
from __future__ import annotations
|
|
|
|
import subprocess
|
|
from io import StringIO
|
|
|
|
from rich.console import Console
|
|
|
|
from whichllm.hardware import intel
|
|
from whichllm.hardware.types import GPUInfo, HardwareInfo
|
|
|
|
|
|
def test_detect_intel_gpu_from_lspci(monkeypatch):
|
|
output = (
|
|
'00:02.0 "VGA compatible controller" "Intel Corporation" '
|
|
'"Alder Lake-P GT1 [UHD Graphics]" -r0c "Dell" "Device 0b19"\n'
|
|
)
|
|
|
|
def fake_run(*args, **kwargs):
|
|
return subprocess.CompletedProcess(args[0], 0, stdout=output, stderr="")
|
|
|
|
monkeypatch.setattr(intel.subprocess, "run", fake_run)
|
|
|
|
gpus = intel.detect_intel_gpus()
|
|
|
|
assert len(gpus) == 1
|
|
assert gpus[0].vendor == "intel"
|
|
assert gpus[0].vram_bytes == 0
|
|
assert "UHD Graphics" in gpus[0].name
|
|
|
|
|
|
def test_detect_intel_arc_pro_b70_from_battlemage_g31_lspci(monkeypatch):
|
|
output = (
|
|
'12:00.0 "VGA compatible controller" "Intel Corporation" '
|
|
'"Battlemage G31 [Intel Graphics]"\n'
|
|
)
|
|
|
|
def fake_run(*args, **kwargs):
|
|
return subprocess.CompletedProcess(args[0], 0, stdout=output, stderr="")
|
|
|
|
monkeypatch.setattr(intel.subprocess, "run", fake_run)
|
|
|
|
gpus = intel.detect_intel_gpus()
|
|
|
|
assert len(gpus) == 1
|
|
assert gpus[0].name == "Battlemage G31 [Intel Graphics]"
|
|
assert gpus[0].vendor == "intel"
|
|
assert gpus[0].vram_bytes == 32 * 1024**3
|
|
assert gpus[0].memory_bandwidth_gbps == 608.0
|
|
assert gpus[0].shared_memory is False
|
|
|
|
|
|
def test_detect_intel_gpu_ignores_non_display_lspci(monkeypatch):
|
|
output = '00:00.0 "Host bridge" "Intel Corporation" "Device 4621"\n'
|
|
|
|
def fake_run(*args, **kwargs):
|
|
return subprocess.CompletedProcess(args[0], 0, stdout=output, stderr="")
|
|
|
|
monkeypatch.setattr(intel.subprocess, "run", fake_run)
|
|
monkeypatch.setattr(intel, "_detect_from_sysfs", lambda: [])
|
|
|
|
assert intel.detect_intel_gpus() == []
|
|
|
|
|
|
def test_detect_intel_gpu_from_sysfs_when_lspci_missing(monkeypatch, tmp_path):
|
|
card = tmp_path / "card0" / "device"
|
|
card.mkdir(parents=True)
|
|
(card / "vendor").write_text("0x8086\n")
|
|
(card / "uevent").write_text("PCI_SLOT_NAME=0000:00:02.0\n")
|
|
|
|
monkeypatch.setattr(intel, "_detect_from_lspci", lambda: [])
|
|
original_sysfs = intel._detect_from_sysfs
|
|
monkeypatch.setattr(intel, "_detect_from_sysfs", lambda: original_sysfs(tmp_path))
|
|
|
|
gpus = intel.detect_intel_gpus()
|
|
|
|
assert len(gpus) == 1
|
|
assert gpus[0].vendor == "intel"
|
|
assert gpus[0].vram_bytes == 0
|
|
assert gpus[0].name == "Intel Integrated Graphics"
|
|
|
|
|
|
def test_detect_intel_arc_pro_b70_from_sysfs_device_id(monkeypatch, tmp_path):
|
|
card = tmp_path / "card0" / "device"
|
|
card.mkdir(parents=True)
|
|
(card / "vendor").write_text("0x8086\n")
|
|
(card / "device").write_text("0xe223\n")
|
|
|
|
monkeypatch.setattr(intel, "_detect_from_lspci", lambda: [])
|
|
original_sysfs = intel._detect_from_sysfs
|
|
monkeypatch.setattr(intel, "_detect_from_sysfs", lambda: original_sysfs(tmp_path))
|
|
|
|
gpus = intel.detect_intel_gpus()
|
|
|
|
assert len(gpus) == 1
|
|
assert gpus[0].name == "Battlemage G31 [Intel Graphics]"
|
|
assert gpus[0].vram_bytes == 32 * 1024**3
|
|
assert gpus[0].memory_bandwidth_gbps == 608.0
|
|
assert gpus[0].shared_memory is False
|
|
|
|
|
|
def test_display_intel_shared_memory_without_zero_kb(monkeypatch):
|
|
from whichllm.output import _console as console_mod
|
|
from whichllm.output import display as display_mod
|
|
|
|
buf = StringIO()
|
|
monkeypatch.setattr(console_mod, "console", Console(file=buf, force_terminal=False))
|
|
|
|
display_mod.display_hardware(
|
|
HardwareInfo(
|
|
gpus=[
|
|
GPUInfo(
|
|
name="Alder Lake-P GT1 [UHD Graphics]",
|
|
vendor="intel",
|
|
vram_bytes=0,
|
|
shared_memory=True,
|
|
)
|
|
],
|
|
cpu_name="CPU",
|
|
cpu_cores=8,
|
|
ram_bytes=16 * 1024**3,
|
|
disk_free_bytes=100 * 1024**3,
|
|
os="linux",
|
|
)
|
|
)
|
|
|
|
output = buf.getvalue()
|
|
assert "shared memory" in output
|
|
assert "0 KB" not in output
|