"""Smoke tests for HardwareProbe — runs on any machine."""
from stressllm.probe import HardwareProbe


def test_probe_returns_all_keys():
    probe = HardwareProbe()
    stats = probe.get_stats()
    assert "ram" in stats
    assert "cpu" in stats
    assert "vram" in stats
    assert "gpu_temp" in stats
    probe.shutdown()


def test_system_summary_returns_all_keys():
    probe = HardwareProbe()
    summary = probe.get_system_summary()
    assert "gpu_name" in summary
    assert "vram_total" in summary
    assert "ram_total" in summary
    assert "gpu_enabled" in summary
    probe.shutdown()
