Spaces:
Running
Running
File size: 787 Bytes
13a295c | 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 | import pytest
from mlplo.app import load_model_info, MODE_PRESETS
from pathlib import Path
import json
def test_load_model_info_fallback():
info = load_model_info("facebook/bart-large-xsum")
assert "Fallback Model" in info
assert "facebook/bart-large-xsum" in info
def test_load_model_info_local(tmp_path):
metrics_dir = tmp_path / "metrics"
metrics_dir.mkdir()
metrics_file = metrics_dir / "test_metrics.json"
metrics_file.write_text(json.dumps({"test_rouge1": 0.45, "test_rougeL": 0.40}))
info = load_model_info(str(tmp_path))
assert "Local Checkpoint" in info
assert "ROUGE-1" in info
assert "0.45" in info
def test_mode_presets():
assert "Quick Pulse" in MODE_PRESETS
assert "max_new_tokens" in MODE_PRESETS["Quick Pulse"]
|