mirror of
https://github.com/NVIDIA-NeMo/DataDesigner
synced 2026-05-24 09:48:29 +00:00
8b79b21298
Preserves tree from previous docs-website head: 5e47d33ea8. This branch is a CI-managed publish artifact like gh-pages; source provenance is tracked in commit messages rather than Git ancestry.
41 lines
1.6 KiB
Python
41 lines
1.6 KiB
Python
# SPDX-FileCopyrightText: Copyright (c) 2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
|
|
# SPDX-License-Identifier: Apache-2.0
|
|
|
|
from __future__ import annotations
|
|
|
|
from unittest.mock import patch
|
|
|
|
import data_designer.cli.runtime as runtime_mod
|
|
|
|
|
|
def test_ensure_cli_default_model_settings_attempts_default_setup() -> None:
|
|
"""CLI bootstrap delegates to default setup when the CLI starts."""
|
|
with (
|
|
patch("data_designer.cli.runtime.print_warning") as mock_print_warning,
|
|
patch("data_designer.cli.runtime.resolve_seed_default_model_settings") as mock_resolve,
|
|
):
|
|
runtime_mod.ensure_cli_default_model_settings()
|
|
|
|
mock_resolve.assert_called_once_with()
|
|
mock_print_warning.assert_not_called()
|
|
|
|
|
|
def test_ensure_cli_default_model_settings_warns_and_continues() -> None:
|
|
"""CLI bootstrap prints an actionable warning when setup fails."""
|
|
with (
|
|
patch("data_designer.cli.runtime.print_warning") as mock_print_warning,
|
|
patch(
|
|
"data_designer.cli.runtime.resolve_seed_default_model_settings",
|
|
side_effect=RuntimeError("boom"),
|
|
) as mock_resolve,
|
|
):
|
|
runtime_mod.ensure_cli_default_model_settings()
|
|
|
|
mock_resolve.assert_called_once_with()
|
|
mock_print_warning.assert_called_once()
|
|
warning = mock_print_warning.call_args[0][0]
|
|
assert "Could not initialize default model providers and model configs automatically." in warning
|
|
assert "The command will continue." in warning
|
|
assert "boom" in warning
|
|
assert "data-designer config providers" in warning
|
|
assert "data-designer config models" in warning
|