File size: 895 Bytes
f526878 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 |
import sys
from pathlib import Path
import pytest
from typer.testing import CliRunner
from ca.cli import app
runner = CliRunner()
def test_train_group_help():
result = runner.invoke(app, ["train", "--help"])
assert result.exit_code == 0
assert "qlora" in result.stdout
def test_train_validate_missing_dataset(tmp_path: Path):
missing_dir = tmp_path / "missing"
result = runner.invoke(app, ["train", "validate", "--dataset-dir", str(missing_dir)])
assert result.exit_code != 0
def test_doctor_training_check_runs(tmp_path: Path):
dataset_root = tmp_path / "ds"
(dataset_root / "data").mkdir(parents=True, exist_ok=True)
(dataset_root / "eval").mkdir(parents=True, exist_ok=True)
result = runner.invoke(app, ["doctor", "--check-training", "--dataset-dir", str(dataset_root)])
assert result.exit_code == 0
assert "Dataset" in result.stdout
|