Fix tests

Author: nabinchha

src/data_designer/cli/ui.py

@@ -14,6 +14,7 @@
 from prompt_toolkit.layout.controls import FormattedTextControl
 from prompt_toolkit.styles import Style
 from rich.console import Console
+from rich.padding import Padding
 from rich.panel import Panel
 
 from data_designer.config.utils.constants import RICH_CONSOLE_THEME, NordColor
@@ -431,7 +432,6 @@ def _print_with_padding(content: str | Panel) -> None:
     Args:
         content: Content to print (string or Panel)
     """
-    from rich.padding import Padding
 
     padding = " " * LEFT_PADDING
     if isinstance(content, Panel):

tests/cli/commands/test_list.py tests/cli/commands/test_list_command.pytests/cli/commands/test_list.py renamed to tests/cli/commands/test_list_command.py

@@ -7,8 +7,6 @@
 
 from data_designer.cli.commands.list import display_models, display_providers, list_command
 from data_designer.cli.constants import DEFAULT_CONFIG_DIR
-from data_designer.cli.repositories.model_repository import ModelRepository
-from data_designer.cli.repositories.provider_repository import ProviderRepository
 
 
 @patch("data_designer.cli.commands.list.display_providers")
@@ -23,18 +21,18 @@ def test_list_command(mock_display_models, mock_display_providers):
 
 
 @patch("data_designer.cli.commands.list.console.print")
-def test_display_providers(mock_console_print):
+def test_display_providers(mock_console_print, stub_provider_service):
     """Test display providers."""
-    display_providers(ProviderRepository(DEFAULT_CONFIG_DIR))
+    display_providers(stub_provider_service.repository)
     mock_console_print.call_count > 1
     assert isinstance(mock_console_print.call_args_list[0][0][0], Table)
     mock_console_print.call_args_list[0][0][0].title == "Model Providers"
 
 
 @patch("data_designer.cli.commands.list.console.print")
-def test_display_models(mock_console_print):
+def test_display_models(mock_console_print, stub_model_service):
     """Test display models."""
-    display_models(ModelRepository(DEFAULT_CONFIG_DIR))
+    display_models(stub_model_service.repository)
     mock_console_print.call_count > 1
     assert isinstance(mock_console_print.call_args_list[0][0][0], Table)
     mock_console_print.call_args_list[0][0][0].title == "Model Configurations"