hud-python 0.3.5__py3-none-any.whl → 0.4.0__py3-none-any.whl

hud/utils/tests/test_init.py

@@ -1,21 +1,17 @@
-"""Test utils package imports."""
-
-from __future__ import annotations
-
-
-def test_utils_imports():
-    """Test that utils package can be imported."""
-    import hud.utils
-
-    # Check that the module exists
-    assert hud.utils is not None
-
-    # Try importing submodules
-    from hud.utils import agent, common, config, misc, progress, telemetry
-
-    assert agent is not None
-    assert common is not None
-    assert config is not None
-    assert misc is not None
-    assert progress is not None
-    assert telemetry is not None
+"""Test utils package imports."""
+
+from __future__ import annotations
+
+
+def test_utils_imports():
+    """Test that utils package can be imported."""
+    import hud.utils
+
+    # Check that the module exists
+    assert hud.utils is not None
+
+    # Try importing submodules
+    from hud.utils import progress, telemetry
+
+    assert progress is not None
+    assert telemetry is not None

hud/utils/tests/test_progress.py

@@ -1,225 +1,261 @@
-"""Tests for the progress tracking utilities."""
-
-from __future__ import annotations
-
-import pytest
-
-from hud.utils.progress import StepProgressTracker
-
-
-@pytest.fixture
-def tracker():
-    return StepProgressTracker(total_tasks=2, max_steps_per_task=10)
-
-
-def test_invalid_inputs_init():
-    with pytest.raises(ValueError, match="total_tasks must be positive"):
-        StepProgressTracker(total_tasks=0, max_steps_per_task=10)
-
-    with pytest.raises(ValueError, match="max_steps_per_task must be positive"):
-        StepProgressTracker(total_tasks=5, max_steps_per_task=0)
-
-
-def test_start_task(tracker):
-    assert tracker.start_time is None
-    assert tracker._tasks_started == 0
-
-    tracker.start_task("task1")
-
-    assert tracker.start_time is not None
-    assert tracker._tasks_started == 1
-    assert tracker._task_steps["task1"] == 0
-    assert not tracker._finished_tasks["task1"]
-
-    tracker.start_task("task2")
-    assert tracker._tasks_started == 2
-    assert tracker._task_steps["task2"] == 0
-    assert not tracker._finished_tasks["task2"]
-
-
-def test_increment_step(tracker):
-    tracker.start_task("task1")
-    assert tracker.current_total_steps == 0
-
-    tracker.increment_step("task1")
-    assert tracker._task_steps["task1"] == 1
-    assert tracker.current_total_steps == 1
-
-    tracker.increment_step("task1")
-    tracker.increment_step("task1")
-    assert tracker._task_steps["task1"] == 3
-    assert tracker.current_total_steps == 3
-
-    tracker.start_task("task2")
-    tracker.increment_step("task2")
-    assert tracker._task_steps["task2"] == 1
-    assert tracker.current_total_steps == 4
-
-    tracker.finish_task("task1")
-    initial_steps = tracker.current_total_steps
-    tracker.increment_step("task1")
-    assert tracker.current_total_steps == initial_steps
-
-    for _ in range(15):
-        tracker.increment_step("task2")
-    assert tracker._task_steps["task2"] <= tracker.max_steps_per_task
-
-
-def test_finish_task(tracker):
-    tracker.start_task("task1")
-    tracker.start_task("task2")
-
-    tracker.increment_step("task1")
-    tracker.increment_step("task1")
-    initial_steps = tracker._task_steps["task1"]
-
-    tracker.finish_task("task1")
-
-    assert tracker._finished_tasks["task1"]
-    assert tracker._tasks_finished == 1
-    assert tracker._task_steps["task1"] == tracker.max_steps_per_task
-    assert tracker.current_total_steps > initial_steps
-
-    current_steps = tracker.current_total_steps
-    tracker.finish_task("task1")
-    assert tracker._tasks_finished == 1
-    assert tracker.current_total_steps == current_steps
-
-
-def test_get_progress(tracker):
-    steps, total, percentage = tracker.get_progress()
-    assert steps == 0
-    assert total == tracker.total_potential_steps
-    assert percentage == 0.0
-
-    tracker.start_task("task1")
-    tracker.increment_step("task1")
-    steps, total, percentage = tracker.get_progress()
-    assert steps == 1
-    assert total == tracker.total_potential_steps
-    assert percentage == (1 / tracker.total_potential_steps) * 100
-
-    tracker.finish_task("task1")
-    steps, total, percentage = tracker.get_progress()
-    assert steps == tracker.max_steps_per_task
-    assert total == tracker.total_potential_steps
-    assert percentage == (tracker.max_steps_per_task / tracker.total_potential_steps) * 100
-
-    tracker.start_task("task2")
-    tracker.finish_task("task2")
-    steps, total, percentage = tracker.get_progress()
-    assert steps == tracker.total_potential_steps
-    assert percentage == 100.0
-
-
-def test_get_stats_no_progress(tracker, mocker):
-    rate, eta = tracker.get_stats()
-    assert rate == 0.0
-    assert eta is None
-
-    mocker.patch("time.monotonic", return_value=100.0)
-    tracker.start_task("task1")
-
-    mocker.patch("time.monotonic", return_value=100.0)
-    rate, eta = tracker.get_stats()
-    assert rate == 0.0
-    assert eta is None
-
-
-def test_get_stats_with_progress(mocker):
-    mock_time = mocker.patch("time.monotonic")
-    mock_time.return_value = 100.0
-
-    tracker = StepProgressTracker(total_tasks=1, max_steps_per_task=10)
-    tracker.start_task("task1")
-
-    mock_time.return_value = 160.0
-    for _ in range(5):
-        tracker.increment_step("task1")
-
-    rate, eta = tracker.get_stats()
-
-    assert rate == pytest.approx(5.0)
-    assert eta == pytest.approx(60.0)
-
-    for _ in range(5):
-        tracker.increment_step("task1")
-
-    rate, eta = tracker.get_stats()
-    assert rate == pytest.approx(10.0)
-    assert eta == pytest.approx(0.0)
-
-
-def test_is_finished(tracker):
-    assert not tracker.is_finished()
-
-    tracker.start_task("task1")
-    tracker.finish_task("task1")
-    assert not tracker.is_finished()
-
-    tracker.start_task("task2")
-    tracker.finish_task("task2")
-    assert tracker.is_finished()
-
-
-def test_display(tracker, mocker):
-    mock_time = mocker.patch("time.monotonic")
-    mock_time.return_value = 100.0
-    tracker.start_task("task1")
-
-    mock_time.return_value = 130.0
-    tracker.increment_step("task1")
-    tracker.increment_step("task1")
-
-    display_str = tracker.display()
-
-    assert "%" in display_str
-    assert "2/20" in display_str
-    assert "0:30" in display_str
-    assert "steps/min" in display_str
-
-    tracker.finish_task("task1")
-    display_str = tracker.display()
-    assert "10/20" in display_str
-
-    tracker.start_task("task2")
-    tracker.finish_task("task2")
-    display_str = tracker.display()
-    assert "100%" in display_str
-    assert "20/20" in display_str
-
-
-def test_complex_workflow():
-    tracker = StepProgressTracker(total_tasks=5, max_steps_per_task=20)
-
-    for i in range(5):
-        tracker.start_task(f"task{i}")
-
-    for _ in range(10):
-        tracker.increment_step("task0")
-
-    for _ in range(5):
-        tracker.increment_step("task1")
-
-    tracker.finish_task("task2")
-
-    for _ in range(15):
-        tracker.increment_step("task3")
-
-    tracker.finish_task("task3")
-
-    steps, total, percentage = tracker.get_progress()
-    expected_steps = 10 + 5 + 20 + 20 + 0
-    assert steps == expected_steps
-    assert total == 5 * 20
-    assert percentage == (expected_steps / total) * 100
-
-    assert tracker._tasks_finished == 2
-    assert not tracker.is_finished()
-
-    tracker.finish_task("task0")
-    tracker.finish_task("task1")
-    tracker.finish_task("task4")
-
-    assert tracker.is_finished()
-    assert tracker.get_progress()[2] == 100.0
+"""Tests for the progress tracking utilities."""
+
+from __future__ import annotations
+
+import pytest
+
+from hud.utils.progress import StepProgressTracker
+
+
+@pytest.fixture
+def tracker():
+    return StepProgressTracker(total_tasks=2, max_steps_per_task=10)
+
+
+def test_invalid_inputs_init():
+    with pytest.raises(ValueError, match="total_tasks must be positive"):
+        StepProgressTracker(total_tasks=0, max_steps_per_task=10)
+
+    with pytest.raises(ValueError, match="max_steps_per_task must be positive"):
+        StepProgressTracker(total_tasks=5, max_steps_per_task=0)
+
+
+def test_start_task(tracker):
+    assert tracker.start_time is None
+    assert tracker._tasks_started == 0
+
+    tracker.start_task("task1")
+
+    assert tracker.start_time is not None
+    assert tracker._tasks_started == 1
+    assert tracker._task_steps["task1"] == 0
+    assert not tracker._finished_tasks["task1"]
+
+    tracker.start_task("task2")
+    assert tracker._tasks_started == 2
+    assert tracker._task_steps["task2"] == 0
+    assert not tracker._finished_tasks["task2"]
+
+
+def test_increment_step(tracker):
+    tracker.start_task("task1")
+    assert tracker.current_total_steps == 0
+
+    tracker.increment_step("task1")
+    assert tracker._task_steps["task1"] == 1
+    assert tracker.current_total_steps == 1
+
+    tracker.increment_step("task1")
+    tracker.increment_step("task1")
+    assert tracker._task_steps["task1"] == 3
+    assert tracker.current_total_steps == 3
+
+    tracker.start_task("task2")
+    tracker.increment_step("task2")
+    assert tracker._task_steps["task2"] == 1
+    assert tracker.current_total_steps == 4
+
+    tracker.finish_task("task1")
+    initial_steps = tracker.current_total_steps
+    tracker.increment_step("task1")
+    assert tracker.current_total_steps == initial_steps
+
+    for _ in range(15):
+        tracker.increment_step("task2")
+    assert tracker._task_steps["task2"] <= tracker.max_steps_per_task
+
+
+def test_finish_task(tracker):
+    tracker.start_task("task1")
+    tracker.start_task("task2")
+
+    tracker.increment_step("task1")
+    tracker.increment_step("task1")
+    initial_steps = tracker._task_steps["task1"]
+
+    tracker.finish_task("task1")
+
+    assert tracker._finished_tasks["task1"]
+    assert tracker._tasks_finished == 1
+    assert tracker._task_steps["task1"] == tracker.max_steps_per_task
+    assert tracker.current_total_steps > initial_steps
+
+    current_steps = tracker.current_total_steps
+    tracker.finish_task("task1")
+    assert tracker._tasks_finished == 1
+    assert tracker.current_total_steps == current_steps
+
+
+def test_get_progress(tracker):
+    steps, total, percentage = tracker.get_progress()
+    assert steps == 0
+    assert total == tracker.total_potential_steps
+    assert percentage == 0.0
+
+    tracker.start_task("task1")
+    tracker.increment_step("task1")
+    steps, total, percentage = tracker.get_progress()
+    assert steps == 1
+    assert total == tracker.total_potential_steps
+    assert percentage == (1 / tracker.total_potential_steps) * 100
+
+    tracker.finish_task("task1")
+    steps, total, percentage = tracker.get_progress()
+    assert steps == tracker.max_steps_per_task
+    assert total == tracker.total_potential_steps
+    assert percentage == (tracker.max_steps_per_task / tracker.total_potential_steps) * 100
+
+    tracker.start_task("task2")
+    tracker.finish_task("task2")
+    steps, total, percentage = tracker.get_progress()
+    assert steps == tracker.total_potential_steps
+    assert percentage == 100.0
+
+
+def test_get_stats_no_progress(tracker):
+    from unittest.mock import patch
+
+    rate, eta = tracker.get_stats()
+    assert rate == 0.0
+    assert eta is None
+
+    with patch("time.monotonic", return_value=100.0):
+        tracker.start_task("task1")
+
+        rate, eta = tracker.get_stats()
+        assert rate == 0.0
+        assert eta is None
+
+
+def test_get_stats_with_progress():
+    from unittest.mock import patch
+
+    with patch("time.monotonic") as mock_time:
+        mock_time.return_value = 100.0
+
+        tracker = StepProgressTracker(total_tasks=1, max_steps_per_task=10)
+        tracker.start_task("task1")
+
+        mock_time.return_value = 160.0
+        for _ in range(5):
+            tracker.increment_step("task1")
+
+        rate, eta = tracker.get_stats()
+
+        assert rate == pytest.approx(5.0)
+        assert eta == pytest.approx(60.0)
+
+        for _ in range(5):
+            tracker.increment_step("task1")
+
+        rate, eta = tracker.get_stats()
+        assert rate == pytest.approx(10.0)
+        assert eta == pytest.approx(0.0)
+
+
+def test_is_finished(tracker):
+    assert not tracker.is_finished()
+
+    tracker.start_task("task1")
+    tracker.finish_task("task1")
+    assert not tracker.is_finished()
+
+    tracker.start_task("task2")
+    tracker.finish_task("task2")
+    assert tracker.is_finished()
+
+
+def test_display(tracker):
+    from unittest.mock import patch
+
+    with patch("time.monotonic") as mock_time:
+        mock_time.return_value = 100.0
+        tracker.start_task("task1")
+
+        mock_time.return_value = 130.0
+        tracker.increment_step("task1")
+        tracker.increment_step("task1")
+
+        display_str = tracker.display()
+
+        assert "%" in display_str
+        assert "2/20" in display_str
+        assert "0:30" in display_str
+        assert "steps/min" in display_str
+
+        tracker.finish_task("task1")
+        display_str = tracker.display()
+        assert "10/20" in display_str
+
+        tracker.start_task("task2")
+        tracker.finish_task("task2")
+        display_str = tracker.display()
+        assert "100%" in display_str
+        assert "20/20" in display_str
+
+
+def test_complex_workflow():
+    tracker = StepProgressTracker(total_tasks=5, max_steps_per_task=20)
+
+    for i in range(5):
+        tracker.start_task(f"task{i}")
+
+    for _ in range(10):
+        tracker.increment_step("task0")
+
+    for _ in range(5):
+        tracker.increment_step("task1")
+
+    tracker.finish_task("task2")
+
+    for _ in range(15):
+        tracker.increment_step("task3")
+
+    tracker.finish_task("task3")
+
+    steps, total, percentage = tracker.get_progress()
+    expected_steps = 10 + 5 + 20 + 20 + 0
+    assert steps == expected_steps
+    assert total == 5 * 20
+    assert percentage == (expected_steps / total) * 100
+
+    assert tracker._tasks_finished == 2
+    assert not tracker.is_finished()
+
+    tracker.finish_task("task0")
+    tracker.finish_task("task1")
+    tracker.finish_task("task4")
+
+    assert tracker.is_finished()
+    assert tracker.get_progress()[2] == 100.0
+
+
+def test_display_eta_when_finished(tracker):
+    from unittest.mock import patch
+
+    """Test that ETA shows 0:00 when progress is finished."""
+
+    with patch("time.monotonic") as mock_time:
+        mock_time.return_value = 100.0
+
+        # Start and complete all tasks
+        tracker.start_task("task1")
+        for _ in range(10):
+            tracker.increment_step("task1")
+        tracker.finish_task("task1")
+
+        tracker.start_task("task2")
+        for _ in range(10):
+            tracker.increment_step("task2")
+        tracker.finish_task("task2")
+
+        # Some time has passed
+        mock_time.return_value = 120.0
+
+        display = tracker.display()
+
+        # When finished, ETA should be 0:00 (not ??:??)
+        assert tracker.is_finished()
+        assert "0:00" in display
+        assert "100%" in display
+        assert "20/20" in display

hud/utils/tests/test_telemetry.py

@@ -1,37 +1,82 @@
-from __future__ import annotations
-
-from hud.utils.telemetry import stream
-
-
-def test_stream():
-    html_content = stream("https://example.com")
-    assert html_content is not None
-    assert "<div style=" in html_content
-    assert 'src="https://example.com"' in html_content
-
-
-def test_display_screenshot():
-    from hud.utils.telemetry import display_screenshot
-
-    # This is a simple 1x1 transparent PNG image in base64 format
-    base64_image = (
-        "iVBORw0KGgoAAAANSUhEUgAAAAEAAAABCAYAAAAfFcSJAAAADUlEQVR42mP8z8BQDwAEhQGAhKmMIQ"
-        "AAABJRU5ErkJggg=="
-    )
-
-    html_content = display_screenshot(base64_image)
-    assert html_content is not None
-    assert "<div style=" in html_content
-    assert "width: 960px" in html_content
-    assert "height: 540px" in html_content
-    assert f"data:image/png;base64,{base64_image}" in html_content
-
-    # Test with custom dimensions
-    custom_html = display_screenshot(base64_image, width=800, height=600)
-    assert "width: 800px" in custom_html
-    assert "height: 600px" in custom_html
-
-    # Test with data URI already included
-    data_uri = f"data:image/png;base64,{base64_image}"
-    uri_html = display_screenshot(data_uri)
-    assert data_uri in uri_html
+from __future__ import annotations
+
+from unittest.mock import patch
+
+from hud.utils.telemetry import stream
+
+
+def test_stream():
+    html_content = stream("https://example.com")
+    assert html_content is not None
+    assert "<div style=" in html_content
+    assert 'src="https://example.com"' in html_content
+
+
+def test_stream_with_display_exception():
+    """Test stream when IPython display raises an exception."""
+    with (
+        patch("IPython.display.display", side_effect=Exception("Display error")),
+        patch("hud.utils.telemetry.logger") as mock_logger,
+    ):
+        html_content = stream("https://example.com")
+
+        # Should still return the HTML content
+        assert html_content is not None
+        assert 'src="https://example.com"' in html_content
+
+        # Should log the warning
+        mock_logger.warning.assert_called_once()
+        args = mock_logger.warning.call_args[0]
+        assert "Display error" in str(args[0])
+
+
+def test_display_screenshot():
+    from hud.utils.telemetry import display_screenshot
+
+    # This is a simple 1x1 transparent PNG image in base64 format
+    base64_image = (
+        "iVBORw0KGgoAAAANSUhEUgAAAAEAAAABCAYAAAAfFcSJAAAADUlEQVR42mP8z8BQDwAEhQGAhKmMIQ"
+        "AAABJRU5ErkJggg=="
+    )
+
+    html_content = display_screenshot(base64_image)
+    assert html_content is not None
+    assert "<div style=" in html_content
+    assert "width: 960px" in html_content
+    assert "height: 540px" in html_content
+    assert f"data:image/png;base64,{base64_image}" in html_content
+
+    # Test with custom dimensions
+    custom_html = display_screenshot(base64_image, width=800, height=600)
+    assert "width: 800px" in custom_html
+    assert "height: 600px" in custom_html
+
+    # Test with data URI already included
+    data_uri = f"data:image/png;base64,{base64_image}"
+    uri_html = display_screenshot(data_uri)
+    assert data_uri in uri_html
+
+
+def test_display_screenshot_with_exception():
+    """Test display_screenshot when IPython display raises an exception."""
+    from hud.utils.telemetry import display_screenshot
+
+    base64_image = (
+        "iVBORw0KGgoAAAANSUhEUgAAAAEAAAABCAYAAAAfFcSJAAAADUlEQVR42mP8z8BQDwAEhQGAhKmMIQ"
+        "AAABJRU5ErkJggg=="
+    )
+
+    with (
+        patch("IPython.display.display", side_effect=Exception("Display error")),
+        patch("hud.utils.telemetry.logger") as mock_logger,
+    ):
+        html_content = display_screenshot(base64_image)
+
+        # Should still return the HTML content
+        assert html_content is not None
+        assert f"data:image/png;base64,{base64_image}" in html_content
+
+        # Should log the warning
+        mock_logger.warning.assert_called_once()
+        args = mock_logger.warning.call_args[0]
+        assert "Display error" in str(args[0])

hud/utils/tests/test_version.py

@@ -1,8 +1,8 @@
-from __future__ import annotations
-
-
-def test_import():
-    """Test that the package can be imported."""
-    import hud
-
-    assert hud.__version__ == "0.3.5"
+from __future__ import annotations
+
+
+def test_import():
+    """Test that the package can be imported."""
+    import hud
+
+    assert hud.__version__ == "0.4.0"