mirror of
https://github.com/meta-llama/llama-stack.git
synced 2025-10-04 04:04:14 +00:00
45 lines
1.2 KiB
Python
45 lines
1.2 KiB
Python
# Copyright (c) Meta Platforms, Inc. and affiliates.
|
|
# All rights reserved.
|
|
#
|
|
# This source code is licensed under the terms described in the LICENSE file in
|
|
# the root directory of this source tree.
|
|
|
|
import argparse
|
|
from io import StringIO
|
|
from unittest.mock import patch
|
|
|
|
from llama_stack.cli.stack._show import (
|
|
run_stack_show_command,
|
|
)
|
|
|
|
|
|
def test_stack_show_basic():
|
|
args = argparse.Namespace(
|
|
config=None,
|
|
distro=None,
|
|
env_name="test-env",
|
|
providers="inference=remote::ollama",
|
|
)
|
|
|
|
with patch("sys.stdout", new_callable=StringIO) as mock_stdout:
|
|
run_stack_show_command(args)
|
|
output = mock_stdout.getvalue()
|
|
|
|
assert "# Dependencies for test-env" in output
|
|
assert "uv pip install" in output
|
|
|
|
|
|
def test_stack_show_with_distro():
|
|
args = argparse.Namespace(
|
|
config=None,
|
|
distro="starter",
|
|
env_name=None,
|
|
providers=None,
|
|
)
|
|
|
|
with patch("sys.stdout", new_callable=StringIO) as mock_stdout:
|
|
run_stack_show_command(args)
|
|
output = mock_stdout.getvalue()
|
|
|
|
assert "# Dependencies for starter" in output
|
|
assert "uv pip install" in output
|