mirror of
https://github.com/NVIDIA-NeMo/DataDesigner
synced 2026-05-24 09:48:29 +00:00
7cba674a8d
* split person samplers * pass catalogs * remove locale validation on dataset-based person sampler * use parameters in sql execution * not using that * update tests * add temp assets path * update error when too few samples exist * update error message name * add expected params to test
18 lines
877 B
Python
18 lines
877 B
Python
# SPDX-FileCopyrightText: Copyright (c) 2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
|
|
# SPDX-License-Identifier: Apache-2.0
|
|
|
|
from unittest.mock import MagicMock, patch
|
|
|
|
from data_designer.cli.commands.providers import providers_command
|
|
from data_designer.cli.controllers.provider_controller import ProviderController
|
|
from data_designer.config.utils.constants import DATA_DESIGNER_HOME
|
|
|
|
|
|
@patch("data_designer.cli.commands.providers.ProviderController")
|
|
def test_providers_command(mock_provider_controller):
|
|
mock_provider_controller_instance = MagicMock(spec=ProviderController)
|
|
mock_provider_controller.return_value = mock_provider_controller_instance
|
|
providers_command()
|
|
mock_provider_controller.assert_called_once()
|
|
mock_provider_controller.call_args[0][0] == DATA_DESIGNER_HOME
|
|
mock_provider_controller_instance.run.assert_called_once()
|