Spaces:
Sleeping
Sleeping
File size: 3,924 Bytes
adb1607 | 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 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 | import sys
import unittest
from pathlib import Path
sys.path.insert(0, str(Path(__file__).resolve().parents[1]))
import main as pitch_main
PROJECT_ROOT = Path(__file__).resolve().parents[2]
LIVE_CASES = [
{
"path": PROJECT_ROOT / "audios-testes" / "teste_teclado.mp4",
"instrumento": "teclado",
"expected_stage": "F#m A D C#sus4 C#",
"categoria": "harmonico",
},
{
"path": PROJECT_ROOT / "audios-testes" / "teclado_bia_ruido.ogg",
"instrumento": "teclado",
"categoria": "harmonico",
"noisy": True,
},
{
"path": PROJECT_ROOT / "audios-testes" / "teste_sax_alto.ogg",
"instrumento": "sax_alto",
"categoria": "melodico",
},
{
"path": PROJECT_ROOT / "audios-testes" / "sax_alto_bia_ruido.ogg",
"instrumento": "sax_alto",
"categoria": "melodico",
},
{
"path": PROJECT_ROOT / "audios-testes" / "teste_violino.ogg",
"instrumento": "violino",
"categoria": "melodico",
},
]
@unittest.skipUnless(
all(case["path"].exists() for case in LIVE_CASES),
"Clips de regressao live nao encontrados",
)
class LiveAudioRegressionTests(unittest.TestCase):
def assert_progression_overlap(
self,
actual_progression: str,
expected_progression: str,
minimum_overlap: int = 2,
) -> None:
actual_tokens = [item for item in str(actual_progression or "").split() if item]
expected_tokens = [item for item in str(expected_progression or "").split() if item]
overlap = len(set(actual_tokens) & set(expected_tokens))
self.assertGreaterEqual(
overlap,
minimum_overlap,
f"sobreposicao live insuficiente entre '{actual_progression}' e '{expected_progression}'",
)
def test_live_harmonic_path_produces_playable_stage_output(self):
for case in LIVE_CASES:
with self.subTest(audio=case["path"].name, instrumento=case["instrumento"]):
resultado = pitch_main.analisar_harmonico(
case["path"],
case["instrumento"],
contexto="live",
)
self.assertEqual(resultado["tipo"], "harmonico")
self.assertTrue(resultado["acorde_atual"], "live precisa devolver acorde_atual")
self.assertTrue(resultado["cifra_palco"], "live precisa devolver cifra_palco")
self.assertIn("diagnostico_harmonico", resultado)
self.assertIn("timings", resultado)
self.assertTrue(resultado["diagnostico_harmonico"]["fast_mode"])
self.assertGreaterEqual(float(resultado.get("beat_confidence", 0.0)), 0.0)
if case["categoria"] == "harmonico":
if case.get("expected_stage"):
self.assert_progression_overlap(resultado["cifra_palco"], case["expected_stage"])
self.assertGreaterEqual(len(resultado.get("acordes_resumo", [])), 1)
if case.get("noisy"):
self.assertGreaterEqual(
float(resultado["diagnostico_harmonico"].get("segment_count", 0)),
1.0,
)
self.assertGreaterEqual(len(resultado.get("current_chord_candidates", [])), 1)
else:
# Para melodicos, o live agora precisa ao menos devolver base harmonica utilizavel.
self.assertGreaterEqual(len(resultado.get("acordes_resumo", [])), 1)
self.assertGreaterEqual(len(resultado.get("acordes_janela_final", [])), 1)
self.assertGreaterEqual(
float(resultado["diagnostico_harmonico"].get("segment_count", 0)),
1.0,
)
if __name__ == "__main__":
unittest.main()
|