mirror of
https://github.com/open-jarvis/OpenJarvis.git
synced 2026-07-28 14:07:55 +00:00
45 lines
1.5 KiB
Python
45 lines
1.5 KiB
Python
"""Tests for the jarvis learning CLI subcommand group."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from click.testing import CliRunner
|
|
|
|
|
|
class TestLearningCLI:
|
|
def test_learning_group_exists(self) -> None:
|
|
from openjarvis.learning.distillation.cli import learning_group
|
|
|
|
runner = CliRunner()
|
|
result = runner.invoke(learning_group, ["--help"])
|
|
assert result.exit_code == 0
|
|
out = result.output.lower()
|
|
assert "learning" in out or "distillation" in out
|
|
|
|
def test_init_subcommand(self) -> None:
|
|
from openjarvis.learning.distillation.cli import learning_group
|
|
|
|
runner = CliRunner()
|
|
result = runner.invoke(learning_group, ["init", "--help"])
|
|
assert result.exit_code == 0
|
|
|
|
def test_run_subcommand(self) -> None:
|
|
from openjarvis.learning.distillation.cli import learning_group
|
|
|
|
runner = CliRunner()
|
|
result = runner.invoke(learning_group, ["run", "--help"])
|
|
assert result.exit_code == 0
|
|
|
|
def test_history_subcommand(self) -> None:
|
|
from openjarvis.learning.distillation.cli import learning_group
|
|
|
|
runner = CliRunner()
|
|
result = runner.invoke(learning_group, ["history", "--help"])
|
|
assert result.exit_code == 0
|
|
|
|
def test_rollback_subcommand(self) -> None:
|
|
from openjarvis.learning.distillation.cli import learning_group
|
|
|
|
runner = CliRunner()
|
|
result = runner.invoke(learning_group, ["rollback", "--help"])
|
|
assert result.exit_code == 0
|