mirror of
https://github.com/elder-plinius/OBLITERATUS.git
synced 2026-08-31 07:00:37 +02:00
fix(ui): refresh GPU memory gauge periodically
This commit is contained in:
@@ -31,3 +31,72 @@ def test_gradio_application_constructs_without_public_listener():
|
||||
assert probe.returncode == 0, probe.stderr
|
||||
assert probe.stdout.strip() == "operator UI constructed without a listener"
|
||||
assert probe.stderr == ""
|
||||
|
||||
|
||||
def test_gpu_memory_refresh_controls_have_one_configurable_polling_loop():
|
||||
"""Exercise the rendered Gradio graph rather than relying on source shape."""
|
||||
pytest.importorskip("gradio", reason="install the locked spaces extra")
|
||||
probe = subprocess.run(
|
||||
[
|
||||
sys.executable,
|
||||
"-c",
|
||||
r'''
|
||||
import asyncio
|
||||
import app
|
||||
|
||||
config = app.demo.get_config_file()
|
||||
components = config["components"]
|
||||
timers = [item for item in components if item["type"] == "timer"]
|
||||
assert len(timers) == 1
|
||||
timer = timers[0]
|
||||
assert timer["props"]["value"] == 30.0
|
||||
assert timer["props"]["active"] is True
|
||||
|
||||
selectors = [
|
||||
item for item in components
|
||||
if "vram-refresh-control" in item["props"].get("elem_classes", [])
|
||||
]
|
||||
assert len(selectors) == 1
|
||||
selector = selectors[0]
|
||||
assert selector["type"] == "dropdown"
|
||||
assert selector["props"]["label"] == "GPU refresh"
|
||||
assert selector["props"]["choices"] == [
|
||||
("10 sec", 10.0), ("30 sec", 30.0), ("1 min", 60.0)
|
||||
]
|
||||
assert selector["props"]["value"] == 30.0
|
||||
assert selector["props"]["filterable"] is False
|
||||
assert selector["props"]["allow_custom_value"] is False
|
||||
assert selector["props"]["min_width"] == 96
|
||||
|
||||
timer_ticks = [
|
||||
dependency for dependency in config["dependencies"]
|
||||
if (timer["id"], "tick") in dependency["targets"]
|
||||
]
|
||||
assert len(timer_ticks) == 1
|
||||
assert timer_ticks[0]["queue"] is False
|
||||
assert timer_ticks[0]["outputs"] == [app.vram_display._id]
|
||||
|
||||
interval_changes = [
|
||||
dependency for dependency in config["dependencies"]
|
||||
if (selector["id"], "change") in dependency["targets"]
|
||||
and dependency["outputs"] == [timer["id"]]
|
||||
]
|
||||
assert len(interval_changes) == 1
|
||||
assert interval_changes[0]["queue"] is False
|
||||
|
||||
async def verify_interval_changes():
|
||||
for seconds in (10.0, 30.0, 60.0):
|
||||
result = await app.demo.process_api(
|
||||
interval_changes[0]["id"], [seconds], state=None
|
||||
)
|
||||
assert result["data"] == [seconds]
|
||||
|
||||
asyncio.run(verify_interval_changes())
|
||||
''',
|
||||
],
|
||||
capture_output=True,
|
||||
text=True,
|
||||
timeout=120,
|
||||
check=False,
|
||||
)
|
||||
assert probe.returncode == 0, probe.stdout + probe.stderr
|
||||
|
||||
+22
-1
@@ -4,7 +4,12 @@ from __future__ import annotations
|
||||
|
||||
from types import SimpleNamespace
|
||||
|
||||
from obliteratus.ui_vram import render_vram_html
|
||||
from obliteratus.ui_vram import (
|
||||
DEFAULT_VRAM_REFRESH_INTERVAL,
|
||||
VRAM_REFRESH_CHOICES,
|
||||
render_vram_html,
|
||||
resolve_vram_refresh_interval,
|
||||
)
|
||||
|
||||
|
||||
def _memory(used, reserved, total, name):
|
||||
@@ -79,3 +84,19 @@ def test_vram_html_handles_memory_errors():
|
||||
get_memory_info=lambda _index: (_ for _ in ()).throw(RuntimeError("unavailable")),
|
||||
)
|
||||
assert "Memory: unavailable" in render_vram_html(device)
|
||||
|
||||
device.is_gpu_available = lambda: (_ for _ in ()).throw(RuntimeError("unavailable"))
|
||||
assert "Memory: unavailable" in render_vram_html(device)
|
||||
|
||||
|
||||
def test_vram_refresh_interval_contract():
|
||||
assert VRAM_REFRESH_CHOICES == (
|
||||
("10 sec", 10.0),
|
||||
("30 sec", 30.0),
|
||||
("1 min", 60.0),
|
||||
)
|
||||
assert DEFAULT_VRAM_REFRESH_INTERVAL == 30.0
|
||||
assert resolve_vram_refresh_interval(10.0) == 10.0
|
||||
assert resolve_vram_refresh_interval(30.0) == 30.0
|
||||
assert resolve_vram_refresh_interval(60.0) == 60.0
|
||||
assert resolve_vram_refresh_interval(15.0) == 30.0
|
||||
|
||||
Reference in New Issue
Block a user