74 lines
2.4 KiB
Python
74 lines
2.4 KiB
Python
import tempfile, os
|
|
import numpy as np
|
|
from core.audio_scan import scan_video, load_classifier, default_model_path
|
|
|
|
|
|
def test_scan_video_no_model_returns_empty():
|
|
"""scan_video with no model should return empty list."""
|
|
with tempfile.NamedTemporaryFile(suffix=".wav", delete=False) as vid:
|
|
import soundfile as sf
|
|
sf.write(vid.name, np.random.randn(16000 * 20).astype(np.float32) * 0.1, 16000)
|
|
try:
|
|
regions = scan_video(vid.name, model=None)
|
|
assert regions == []
|
|
finally:
|
|
os.unlink(vid.name)
|
|
|
|
|
|
def test_load_classifier_missing_returns_none():
|
|
assert load_classifier("/no/such/model.joblib") is None
|
|
|
|
|
|
def test_default_model_path_contains_profile():
|
|
path = default_model_path("test_profile")
|
|
assert "test_profile" in path
|
|
assert path.endswith(".joblib")
|
|
|
|
|
|
def test_embed_dim_multi_layer():
|
|
from core.audio_scan import _embed_dim
|
|
# Multi-layer models should report concatenated dimension
|
|
assert _embed_dim("HUBERT_XLARGE_ML") == 5120
|
|
assert _embed_dim("HUBERT_LARGE_ML") == 4096
|
|
assert _embed_dim("HUBERT_BASE_ML") == 3072
|
|
# Single-layer unchanged
|
|
assert _embed_dim("HUBERT_XLARGE") == 1280
|
|
|
|
|
|
def test_ml_config():
|
|
from core.audio_scan import _ml_config
|
|
assert _ml_config("HUBERT_XLARGE") is None
|
|
assert _ml_config("BEATS_ML") is None # BEATS has no ML variant
|
|
base, layers = _ml_config("HUBERT_XLARGE_ML")
|
|
assert base == "HUBERT_XLARGE"
|
|
assert layers == [11, 23, 35, 47]
|
|
base, layers = _ml_config("HUBERT_BASE_ML")
|
|
assert base == "HUBERT_BASE"
|
|
assert layers == [2, 5, 8, 11]
|
|
|
|
|
|
def test_embed_dim_ast():
|
|
from core.audio_scan import _embed_dim
|
|
assert _embed_dim("AST") == 768
|
|
assert _embed_dim("AST_ML") == 3072
|
|
|
|
|
|
def test_embed_dim_eat():
|
|
from core.audio_scan import _embed_dim
|
|
assert _embed_dim("EAT") == 768
|
|
|
|
|
|
def test_db_get_all_export_paths():
|
|
with tempfile.NamedTemporaryFile(suffix=".db", delete=False) as f:
|
|
path = f.name
|
|
try:
|
|
from core.db import ProcessedDB
|
|
db = ProcessedDB(path)
|
|
db.add("a.mp4", 10.0, "/out/a_001.mp4", profile="test")
|
|
db.add("b.mp4", 20.0, "/out/b_001.mp4", profile="test")
|
|
db.add("c.mp4", 30.0, "/out/c_001.mp4", profile="other")
|
|
paths = db.get_all_export_paths("test")
|
|
assert set(paths) == {"/out/a_001.mp4", "/out/b_001.mp4"}
|
|
finally:
|
|
os.unlink(path)
|