Spaces:
Running on Zero
Running on Zero
File size: 1,390 Bytes
9c07a74 | 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 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 | """L2 tests for lyrics LM — generation is mocked at the model boundary.
The real Qwen 2.5 7B model is never loaded in CI. We only verify the prompt
shape, the call boundary to ``_get_lm()``, and the normalisation pass that
lowercases section tags before returning to the caller.
"""
from __future__ import annotations
from unittest.mock import MagicMock
import lyrics_lm as ll
def test_build_system_prompt_includes_tag_format():
sp = ll.build_system_prompt()
low = sp.lower()
assert "[verse" in low
assert "[chorus" in low
def test_generate_lyrics_calls_lm_and_returns_text(monkeypatch):
fake_lm = MagicMock()
fake_lm.generate.return_value = "[verse] x\n[chorus] y\n"
monkeypatch.setattr(ll, "_get_lm", lambda: fake_lm)
out = ll.generate_lyrics(
brief="a song",
structure="intro, verse, chorus, outro",
language="en",
tone="upbeat",
verse_lines=4,
chorus_lines=4,
bridge_lines=2,
rhyme="loose",
temperature=0.85,
top_p=0.9,
top_k=40,
max_new_tokens=200,
seed=42,
)
assert "[verse]" in out
fake_lm.generate.assert_called_once()
def test_normalise_lyrics_lowercases_tags():
norm = ll._normalise(" [Verse 1]\nhello\n[Chorus]\nworld ")
assert "[verse 1]" in norm
assert "[chorus]" in norm
assert "[Verse" not in norm
|