mirror of
https://github.com/meta-llama/llama-stack.git
synced 2025-10-04 20:14:13 +00:00
test: convert tests to use show
Signed-off-by: Charlie Doern <cdoern@redhat.com>
This commit is contained in:
parent
2584c73295
commit
26a490b7fc
16 changed files with 177 additions and 239 deletions
45
tests/unit/distribution/test_stack_show.py
Normal file
45
tests/unit/distribution/test_stack_show.py
Normal file
|
@ -0,0 +1,45 @@
|
|||
# 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
|
Loading…
Add table
Add a link
Reference in a new issue