qbraid-cli 0.10.6__tar.gz → 0.10.7__tar.gz

{qbraid_cli-0.10.6 → qbraid_cli-0.10.7}/.github/workflows/bump-version.yml

@@ -21,7 +21,7 @@ jobs:
       pull-requests: write
     steps:
       - name: Checkout code
-        uses: actions/checkout@v4
+        uses: actions/checkout@v5
         with:
           ref: 'main'
 

{qbraid_cli-0.10.6 → qbraid_cli-0.10.7}/.github/workflows/ci-pr-target.yml

@@ -14,7 +14,7 @@ jobs:
         python-version: ['3.11']
 
     steps:
-      - uses: actions/checkout@v4
+      - uses: actions/checkout@v5
       - name: Set up Python ${{ matrix.python-version }}
         uses: actions/setup-python@v5
         with:
@@ -44,14 +44,14 @@ jobs:
       fail-fast: true
 
     steps:
-      - uses: actions/checkout@v4
+      - uses: actions/checkout@v5
       - name: Set up Python ${{ matrix.python-version }}
         uses: actions/setup-python@v5
         with:
           python-version: ${{ matrix.python-version }}
           cache: pip
       - name: Download built package
-        uses: actions/download-artifact@v4
+        uses: actions/download-artifact@v5
         with:
           name: built-package
           path: dist
@@ -73,7 +73,7 @@ jobs:
           QBRAID_API_KEY: ${{ secrets.QBRAID_API_KEY }}
       - name: Upload coverage to Codecov
         if: matrix.python-version == '3.11' && matrix.os == 'ubuntu-latest'
-        uses: codecov/codecov-action@v5.4.2
+        uses: codecov/codecov-action@v5.4.3
         with:
           token: ${{ secrets.CODECOV_TOKEN }}
           fail_ci_if_error: false

{qbraid_cli-0.10.6 → qbraid_cli-0.10.7}/.github/workflows/docs-pr.yml

@@ -12,7 +12,7 @@ jobs:
       cli_version: ${{ steps.get_version.outputs.version }}
     steps:
       - name: Checkout code
-        uses: actions/checkout@v4
+        uses: actions/checkout@v5
         with:
           ref: main
 
@@ -52,13 +52,13 @@ jobs:
 
     steps:
       - name: Download docs assets
-        uses: actions/download-artifact@v4
+        uses: actions/download-artifact@v5
         with:
           name: api-reference
           path: ./api-reference
 
       - name: Checkout qBraid/docs repo
-        uses: actions/checkout@v4
+        uses: actions/checkout@v5
         with:
           repository: qBraid/docs
           token: ${{ secrets.BOT_TOKEN }}

{qbraid_cli-0.10.6 → qbraid_cli-0.10.7}/.github/workflows/docs.yml

@@ -14,7 +14,7 @@ jobs:
     runs-on: ubuntu-latest
 
     steps:
-    - uses: actions/checkout@v4
+    - uses: actions/checkout@v5
     - name: Set up Python
       uses: actions/setup-python@v5
       with:

{qbraid_cli-0.10.6 → qbraid_cli-0.10.7}/.github/workflows/format.yml

@@ -12,7 +12,7 @@ jobs:
     runs-on: ubuntu-latest
 
     steps:
-      - uses: actions/checkout@v4
+      - uses: actions/checkout@v5
       - name: Set up Python
         uses: actions/setup-python@v5
         with:

{qbraid_cli-0.10.6 → qbraid_cli-0.10.7}/.github/workflows/main.yml

@@ -17,7 +17,7 @@ jobs:
         python-version: ['3.11']
 
     steps:
-      - uses: actions/checkout@v4
+      - uses: actions/checkout@v5
       - name: Set up Python ${{ matrix.python-version }}
         uses: actions/setup-python@v5
         with:
@@ -47,14 +47,14 @@ jobs:
       fail-fast: true
 
     steps:
-      - uses: actions/checkout@v4
+      - uses: actions/checkout@v5
       - name: Set up Python ${{ matrix.python-version }}
         uses: actions/setup-python@v5
         with:
           python-version: ${{ matrix.python-version }}
           cache: pip
       - name: Download built package
-        uses: actions/download-artifact@v4
+        uses: actions/download-artifact@v5
         with:
           name: built-package
           path: dist
@@ -76,7 +76,7 @@ jobs:
           QBRAID_API_KEY: ${{ secrets.QBRAID_API_KEY }}
       - name: Upload coverage to Codecov
         if: matrix.python-version == '3.11' && matrix.os == 'ubuntu-latest'
-        uses: codecov/codecov-action@v5.4.2
+        uses: codecov/codecov-action@v5.4.3
         with:
           token: ${{ secrets.CODECOV_TOKEN }}
           fail_ci_if_error: false

{qbraid_cli-0.10.6 → qbraid_cli-0.10.7}/.github/workflows/pre-release.yml

@@ -9,7 +9,7 @@ jobs:
     environment: publish
     runs-on: ubuntu-latest
     steps:
-      - uses: actions/checkout@v4
+      - uses: actions/checkout@v5
         with:
           fetch-depth: 1
 

{qbraid_cli-0.10.6 → qbraid_cli-0.10.7}/.github/workflows/publish.yml

@@ -9,7 +9,7 @@ jobs:
     environment: publish
     runs-on: ubuntu-latest
     steps:
-      - uses: actions/checkout@v4
+      - uses: actions/checkout@v5
         with:
           fetch-depth: 1
 

{qbraid_cli-0.10.6 → qbraid_cli-0.10.7}/.github/workflows/tag-on-merge.yml

@@ -18,7 +18,7 @@ jobs:
           echo "version=$VERSION" >> $GITHUB_OUTPUT
 
       - name: Checkout code
-        uses: actions/checkout@v4
+        uses: actions/checkout@v5
         with:
           ref: 'main'
 

{qbraid_cli-0.10.6 → qbraid_cli-0.10.7}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: qbraid-cli
-Version: 0.10.6
+Version: 0.10.7
 Summary: Command Line Interface for interacting with all parts of the qBraid platform.
 Author-email: qBraid Development Team <contact@qbraid.com>
 License: Proprietary
@@ -29,7 +29,7 @@ Description-Content-Type: text/markdown
 License-File: LICENSE
 Requires-Dist: typer>=0.12.1
 Requires-Dist: rich>=10.11.0
-Requires-Dist: click
+Requires-Dist: click<=8.1.8
 Requires-Dist: qbraid-core<0.2,>=0.1.41
 Provides-Extra: jobs
 Requires-Dist: amazon-braket-sdk>=1.48.1; extra == "jobs"

{qbraid_cli-0.10.6 → qbraid_cli-0.10.7}/docs/requirements.txt

@@ -1,6 +1,6 @@
 sphinx>=7.2.6,<8.3.0
 sphinx-rtd-theme>=1.3,<3.1
-docutils<0.22
+docutils<0.23
 toml
 build
 m2r

{qbraid_cli-0.10.6 → qbraid_cli-0.10.7}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "qbraid-cli"
-version = "0.10.6"
+version = "0.10.7"
 description = "Command Line Interface for interacting with all parts of the qBraid platform."
 readme = "README.md"
 authors = [{ name = "qBraid Development Team", email = "contact@qbraid.com" }]
@@ -30,7 +30,7 @@ classifiers = [
 dependencies = [
     "typer>=0.12.1",
     "rich>=10.11.0",
-    "click",
+    "click<=8.1.8",
     "qbraid-core>=0.1.41,<0.2",
 ]
 requires-python = ">= 3.9"

{qbraid_cli-0.10.6 → qbraid_cli-0.10.7}/qbraid_cli/_version.py

@@ -17,5 +17,5 @@ __version__: str
 __version_tuple__: VERSION_TUPLE
 version_tuple: VERSION_TUPLE
 
-__version__ = version = '0.1.dev1+ge2da523'
-__version_tuple__ = version_tuple = (0, 1, 'dev1', 'ge2da523')
+__version__ = version = '0.1.dev1+g494c7ce'
+__version_tuple__ = version_tuple = (0, 1, 'dev1', 'g494c7ce')

{qbraid_cli-0.10.6 → qbraid_cli-0.10.7}/qbraid_cli/devices/app.py

@@ -12,7 +12,7 @@ import typer
 from rich.console import Console
 
 from qbraid_cli.devices.validation import validate_provider, validate_status, validate_type
-from qbraid_cli.handlers import run_progress_task
+from qbraid_cli.handlers import print_formatted_data, run_progress_task
 
 devices_app = typer.Typer(help="Manage qBraid quantum devices.", no_args_is_help=True)
 
@@ -29,7 +29,10 @@ def devices_list(  # pylint: disable=too-many-branches
         None,
         "--provider",
         "-p",
-        help="'AWS'|'IBM'|'IonQ'|'Rigetti'|'OQC'|'QuEra'",
+        help=(
+            "'AWS'|'IBM'|'IonQ'|'Rigetti'|'OQC'|'QuEra'|'IQM'|"
+            "'NEC'|'qBraid'|'Azure'|'Pasqal'|'Quantinuum'|'Equal1'"
+        ),
         callback=validate_provider,
     ),
 ) -> None:
@@ -76,5 +79,25 @@ def devices_list(  # pylint: disable=too-many-branches
     console.print(f"\n{msg}", style="italic", justify="left")
 
 
+@devices_app.command(name="get")
+def devices_get(
+    device_id: str = typer.Argument(..., help="The ID of the device to get."),
+    fmt: bool = typer.Option(
+        True, "--no-fmt", help="Disable rich console formatting (output raw data)"
+    ),
+) -> None:
+    """Get a qBraid quantum device."""
+
+    def get_device():
+        from qbraid_core.services.quantum import QuantumClient
+
+        client = QuantumClient()
+        return client.get_device(device_id)
+
+    data: dict[str, Any] = run_progress_task(get_device)
+
+    print_formatted_data(data, fmt)
+
+
 if __name__ == "__main__":
     devices_app()

{qbraid_cli-0.10.6 → qbraid_cli-0.10.7}/qbraid_cli/envs/app.py

@@ -128,6 +128,7 @@ def envs_create(  # pylint: disable=too-many-statements
     env_config.description = environment.get("description")
     env_config.tags = environment.get("tags")
     env_config.kernel_name = environment.get("kernelName")
+    env_config.python_version = python_version
 
     slug_path = env_path / slug
     description = "None" if description == "" else description

{qbraid_cli-0.10.6 → qbraid_cli-0.10.7}/qbraid_cli/files/app.py

@@ -27,10 +27,10 @@ def is_file_less_than_10mb(file_path: Path) -> bool:
     Returns:
         bool: True if the file is less than 10MB, False otherwise.
     """
-    TEN_MB = 10 * 1024 * 1024  # 10 * 1024 KB * 1024 bytes
+    ten_mb = 10485760  # 10 MB in bytes (10 * 1024 * 1024)
 
     try:
-        return file_path.stat().st_size < TEN_MB
+        return file_path.stat().st_size < ten_mb
     except OSError:
         return False
 

{qbraid_cli-0.10.6 → qbraid_cli-0.10.7}/qbraid_cli/handlers.py

@@ -89,6 +89,22 @@ def handle_filesystem_operation(operation: Callable[[], None], path: Path) -> No
         raise QbraidException(f"Failed to save configuration to {path}: {err.strerror}") from err
 
 
+def print_formatted_data(data: Any, fmt: bool = True) -> None:
+    """
+    Print data with optional formatting using rich console.
+
+    Args:
+        data (Any): The data to be printed.
+        fmt (bool): If True, use rich console formatting. If False, use standard print.
+                   Defaults to True.
+    """
+    if fmt:
+        console = Console()
+        console.print(data)
+    else:
+        print(data)
+
+
 def run_progress_task(
     operation: Callable[..., Any],
     *args,

{qbraid_cli-0.10.6 → qbraid_cli-0.10.7}/qbraid_cli/jobs/app.py

@@ -11,7 +11,7 @@ from typing import Any, Callable
 import typer
 from rich.console import Console
 
-from qbraid_cli.handlers import handle_error, run_progress_task
+from qbraid_cli.handlers import handle_error, print_formatted_data, run_progress_task
 from qbraid_cli.jobs.toggle_braket import disable_braket, enable_braket
 from qbraid_cli.jobs.validation import handle_jobs_state, run_progress_get_state, validate_library
 
@@ -107,11 +107,8 @@ def jobs_list(
 
     result: tuple[Any, Callable] = run_progress_task(import_jobs)
     client, process_job_data = result
-    # https://github.com/qBraid/api/issues/644
-    # raw_data = client.search_jobs(query={"numResults": limit})
-    raw_data = client.search_jobs(query={})
+    raw_data = client.search_jobs(query={"resultsPerPage": limit, "page": 0})
     job_data, msg = process_job_data(raw_data)
-    job_data = job_data[:limit]
 
     longest_job_id = max(len(item[0]) for item in job_data)
     spacing = longest_job_id + 5
@@ -148,5 +145,25 @@ def jobs_list(
         handle_error(message="Failed to fetch quantum jobs.")
 
 
+@jobs_app.command(name="get")
+def jobs_get(
+    job_id: str = typer.Argument(..., help="The ID of the job to get."),
+    fmt: bool = typer.Option(
+        True, "--no-fmt", help="Disable rich console formatting (output raw data)"
+    ),
+) -> None:
+    """Get a qBraid Quantum Job."""
+
+    def get_job():
+        from qbraid_core.services.quantum import QuantumClient
+
+        client = QuantumClient()
+        return client.get_job(job_id)
+
+    data: dict[str, Any] = run_progress_task(get_job)
+
+    print_formatted_data(data, fmt)
+
+
 if __name__ == "__main__":
     jobs_app()

{qbraid_cli-0.10.6 → qbraid_cli-0.10.7}/qbraid_cli.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: qbraid-cli
-Version: 0.10.6
+Version: 0.10.7
 Summary: Command Line Interface for interacting with all parts of the qBraid platform.
 Author-email: qBraid Development Team <contact@qbraid.com>
 License: Proprietary
@@ -29,7 +29,7 @@ Description-Content-Type: text/markdown
 License-File: LICENSE
 Requires-Dist: typer>=0.12.1
 Requires-Dist: rich>=10.11.0
-Requires-Dist: click
+Requires-Dist: click<=8.1.8
 Requires-Dist: qbraid-core<0.2,>=0.1.41
 Provides-Extra: jobs
 Requires-Dist: amazon-braket-sdk>=1.48.1; extra == "jobs"

{qbraid_cli-0.10.6 → qbraid_cli-0.10.7}/qbraid_cli.egg-info/SOURCES.txt

@@ -83,6 +83,7 @@ tests/configure/test_configure_prompt_for_config.py
 tests/configure/test_configure_set.py
 tests/configure/test_configure_validate_input.py
 tests/devices/__init__.py
+tests/devices/test_devices_get.py
 tests/devices/test_devices_list.py
 tests/devices/test_devices_validations.py
 tests/envs/__init__.py
@@ -98,6 +99,7 @@ tests/files/test_files_commands.py
 tests/jobs/__init__.py
 tests/jobs/test_jobs_disable.py
 tests/jobs/test_jobs_enable.py
+tests/jobs/test_jobs_get.py
 tests/jobs/test_jobs_list.py
 tests/jobs/test_jobs_state.py
 tests/jobs/test_jobs_toggle_braket_confirm.py

{qbraid_cli-0.10.6 → qbraid_cli-0.10.7}/qbraid_cli.egg-info/requires.txt

@@ -1,6 +1,6 @@
 typer>=0.12.1
 rich>=10.11.0
-click
+click<=8.1.8
 qbraid-core<0.2,>=0.1.41
 
 [dev]

qbraid_cli-0.10.7/tests/devices/test_devices_get.py

@@ -0,0 +1,278 @@
+# Copyright (c) 2024, qBraid Development Team
+# All rights reserved.
+
+"""
+Unit tests for the `devices_get` function in the `qbraid_cli.devices.app` module.
+
+"""
+
+from unittest.mock import MagicMock, patch
+
+import pytest
+import typer
+from qbraid_core.services.quantum.exceptions import QuantumServiceRequestError
+from typer.testing import CliRunner
+
+from qbraid_cli.devices.app import devices_get
+
+runner = CliRunner()
+
+
+@pytest.mark.remote
+def test_devices_get_formatted_output():
+    """Test the `devices_get` function with formatted output (default behavior)."""
+    mock_device_data = {
+        "id": "test-device-123",
+        "name": "Test Quantum Computer",
+        "provider": "AWS",
+        "status": "ONLINE",
+        "type": "QPU",
+        "properties": {"qubits": 5, "connectivity": "all-to-all"},
+    }
+    device_id = "test-device-123"
+
+    with (
+        patch(
+            "qbraid_core.services.quantum.QuantumClient.get_device", return_value=mock_device_data
+        ),
+        patch("rich.console.Console.print") as mock_console_print,
+    ):
+        devices_get(device_id=device_id, fmt=True)
+
+        # Verify the console was called to print the formatted data (last call should be our data)
+        calls = mock_console_print.call_args_list
+        assert len(calls) >= 1
+        # The last call should be our device data
+        assert calls[-1][0][0] == mock_device_data
+
+
+@pytest.mark.remote
+def test_devices_get_raw_output():
+    """Test the `devices_get` function with raw output (fmt=False)."""
+    mock_device_data = {
+        "id": "test-device-456",
+        "name": "Test Simulator",
+        "provider": "IBM",
+        "status": "OFFLINE",
+        "type": "SIMULATOR",
+    }
+    device_id = "test-device-456"
+
+    with (
+        patch(
+            "qbraid_core.services.quantum.QuantumClient.get_device", return_value=mock_device_data
+        ),
+        patch("builtins.print") as mock_print,
+    ):
+        devices_get(device_id=device_id, fmt=False)
+
+        # Verify print was called with the raw data
+        mock_print.assert_called_once_with(mock_device_data)
+
+
+@pytest.mark.remote
+def test_devices_get_default_fmt_behavior():
+    """Test that `devices_get` defaults to formatted output when fmt is not specified."""
+    mock_device_data = {
+        "id": "test-device-789",
+        "name": "Default Test Device",
+        "provider": "IonQ",
+        "status": "RETIRED",
+        "type": "QPU",
+    }
+    device_id = "test-device-789"
+
+    with (
+        patch(
+            "qbraid_core.services.quantum.QuantumClient.get_device", return_value=mock_device_data
+        ),
+        patch("rich.console.Console.print") as mock_console_print,
+    ):
+        devices_get(device_id=device_id)  # fmt defaults to True
+
+        # Verify the console was called to print the formatted data (last call should be our data)
+        calls = mock_console_print.call_args_list
+        assert len(calls) >= 1
+        # The last call should be our device data
+        assert calls[-1][0][0] == mock_device_data
+
+
+@pytest.mark.remote
+def test_devices_get_quantum_client_error():
+    """Test error handling when QuantumClient.get_device raises an exception."""
+    device_id = "invalid-device-id"
+
+    with patch(
+        "qbraid_core.services.quantum.QuantumClient.get_device",
+        side_effect=QuantumServiceRequestError("Device not found"),
+    ):
+        with pytest.raises(typer.Exit) as exc_info:
+            devices_get(device_id=device_id)
+
+        assert exc_info.value.exit_code == 1
+
+
+@pytest.mark.remote
+def test_devices_get_empty_device_data():
+    """Test handling of empty device data."""
+    mock_device_data = {}
+    device_id = "empty-device-id"
+
+    with (
+        patch(
+            "qbraid_core.services.quantum.QuantumClient.get_device", return_value=mock_device_data
+        ),
+        patch("rich.console.Console.print") as mock_console_print,
+    ):
+        devices_get(device_id=device_id, fmt=True)
+
+        # Verify the console was called with empty data (last call should be our data)
+        calls = mock_console_print.call_args_list
+        assert len(calls) >= 1
+        # The last call should be our device data
+        assert calls[-1][0][0] == mock_device_data
+
+
+@pytest.mark.remote
+def test_devices_get_complex_device_data():
+    """Test handling of complex device data with nested structures."""
+    mock_device_data = {
+        "id": "complex-device-123",
+        "name": "Advanced Quantum Computer",
+        "provider": "Rigetti",
+        "status": "ONLINE",
+        "type": "QPU",
+        "specifications": {
+            "qubits": 32,
+            "connectivity": "nearest-neighbor",
+            "gate_fidelity": 0.99,
+            "coherence_time": "100μs",
+        },
+        "calibration": {
+            "last_calibrated": "2023-03-19T12:00:00Z",
+            "next_calibration": "2023-03-26T12:00:00Z",
+            "calibration_data": {
+                "readout_fidelity": 0.95,
+                "single_qubit_gate_fidelity": 0.99,
+                "two_qubit_gate_fidelity": 0.97,
+            },
+        },
+    }
+    device_id = "complex-device-123"
+
+    with (
+        patch(
+            "qbraid_core.services.quantum.QuantumClient.get_device", return_value=mock_device_data
+        ),
+        patch("rich.console.Console.print") as mock_console_print,
+    ):
+        devices_get(device_id=device_id, fmt=True)
+
+        # Verify the console was called with complex data (last call should be our data)
+        calls = mock_console_print.call_args_list
+        assert len(calls) >= 1
+        # The last call should be our device data
+        assert calls[-1][0][0] == mock_device_data
+
+
+@pytest.mark.remote
+def test_devices_get_cli_integration_formatted():
+    """Test the devices get command via CLI with formatted output."""
+    from qbraid_cli.devices.app import devices_app
+
+    mock_device_data = {"id": "cli-test-device", "name": "CLI Test Device", "status": "ONLINE"}
+
+    with (
+        patch(
+            "qbraid_core.services.quantum.QuantumClient.get_device", return_value=mock_device_data
+        ),
+        patch("rich.console.Console.print") as mock_console_print,
+    ):
+        result = runner.invoke(devices_app, ["get", "cli-test-device"])
+
+        assert result.exit_code == 0
+        calls = mock_console_print.call_args_list
+        assert len(calls) >= 1
+        # The last call should be our device data
+        assert calls[-1][0][0] == mock_device_data
+
+
+@pytest.mark.remote
+def test_devices_get_cli_integration_raw():
+    """Test the devices get command via CLI with raw output."""
+    from qbraid_cli.devices.app import devices_app
+
+    mock_device_data = {"id": "cli-raw-device", "name": "CLI Raw Device", "status": "OFFLINE"}
+
+    with (
+        patch(
+            "qbraid_core.services.quantum.QuantumClient.get_device", return_value=mock_device_data
+        ),
+        patch("builtins.print") as mock_print,
+    ):
+        result = runner.invoke(devices_app, ["get", "cli-raw-device", "--no-fmt"])
+
+        assert result.exit_code == 0
+        mock_print.assert_called_once_with(mock_device_data)
+
+
+@pytest.mark.remote
+def test_devices_get_different_provider_devices():
+    """Test getting devices from different providers."""
+    providers = ["AWS", "IBM", "IonQ", "Rigetti", "OQC", "QuEra"]
+
+    for provider in providers:
+        mock_device_data = {
+            "id": f"{provider.lower()}-device-123",
+            "name": f"{provider} Test Device",
+            "provider": provider,
+            "status": "ONLINE",
+            "type": "QPU",
+        }
+        device_id = f"{provider.lower()}-device-123"
+
+        with (
+            patch(
+                "qbraid_core.services.quantum.QuantumClient.get_device",
+                return_value=mock_device_data,
+            ),
+            patch("rich.console.Console.print") as mock_console_print,
+        ):
+            devices_get(device_id=device_id, fmt=True)
+
+            # Verify the console was called with the correct data (last call should be our data)
+            calls = mock_console_print.call_args_list
+            assert len(calls) >= 1
+            # The last call should be our device data
+            assert calls[-1][0][0] == mock_device_data
+
+
+@pytest.mark.remote
+def test_devices_get_different_status_devices():
+    """Test getting devices with different statuses."""
+    statuses = ["ONLINE", "OFFLINE", "RETIRED"]
+
+    for status in statuses:
+        mock_device_data = {
+            "id": f"status-{status.lower()}-device",
+            "name": f"Status {status} Device",
+            "provider": "Test Provider",
+            "status": status,
+            "type": "QPU",
+        }
+        device_id = f"status-{status.lower()}-device"
+
+        with (
+            patch(
+                "qbraid_core.services.quantum.QuantumClient.get_device",
+                return_value=mock_device_data,
+            ),
+            patch("rich.console.Console.print") as mock_console_print,
+        ):
+            devices_get(device_id=device_id, fmt=True)
+
+            # Verify the console was called with the correct data (last call should be our data)
+            calls = mock_console_print.call_args_list
+            assert len(calls) >= 1
+            # The last call should be our device data
+            assert calls[-1][0][0] == mock_device_data