77 lines
3.0 KiB
Python
77 lines
3.0 KiB
Python
"""Tests cho tools/autosample_vsti.py — SF2 writer tự-sinh phải là RIFF/sfbk
|
|
hợp lệ mà sf2utils parse được (không cần VSTi/pedalboard)."""
|
|
import os
|
|
import sys
|
|
import numpy as np
|
|
import pytest
|
|
|
|
from tools.autosample_vsti import write_sf2
|
|
|
|
ROOT = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
|
|
FAKE_VST2 = os.path.join(ROOT, "native_host", "tests", "fake_vst2.dll")
|
|
|
|
sf2utils = pytest.importorskip("sf2utils.sf2parse")
|
|
|
|
|
|
def _sine(note, sr=44100, dur=0.2):
|
|
t = np.arange(int(sr * dur)) / sr
|
|
f = 440.0 * 2 ** ((note - 69) / 12)
|
|
return (np.sin(2 * np.pi * f * t) * 20000).astype(np.int16)
|
|
|
|
|
|
def test_write_sf2_valid_structure(tmp_path):
|
|
out = str(tmp_path / "test.sf2")
|
|
samples = [{"note": n, "frames": _sine(n)} for n in (60, 62, 64)]
|
|
write_sf2(out, samples, sample_rate=44100, name="Test")
|
|
assert os.path.getsize(out) > 100
|
|
with open(out, "rb") as f:
|
|
head = f.read(12)
|
|
assert head[:4] == b"RIFF"
|
|
assert head[8:12] == b"sfbk"
|
|
with open(out, "rb") as f:
|
|
sf2 = sf2utils.Sf2File(f)
|
|
real_samples = [s for s in sf2.samples if s.end > s.start]
|
|
assert len(real_samples) == 3
|
|
assert len(sf2.presets) == 2 # preset + terminator
|
|
assert sf2.presets[0].bank == 0
|
|
assert sf2.presets[0].preset == 0
|
|
assert sf2.presets[0].name == "Test"
|
|
for s in real_samples:
|
|
assert s.end - s.start == 8820
|
|
assert s.sample_rate == 44100
|
|
|
|
|
|
def test_write_sf2_rejects_empty(tmp_path):
|
|
with pytest.raises(ValueError):
|
|
write_sf2(str(tmp_path / "empty.sf2"), [])
|
|
|
|
|
|
def test_write_sf2_odd_name_no_corruption(tmp_path, caplog):
|
|
"""INFO text chunk chan — ten le (vd "Nexus") khong duoc lam mat can
|
|
(sf2utils khong skip pad byte cua odd-size chunk)."""
|
|
import logging
|
|
out = str(tmp_path / "odd.sf2")
|
|
write_sf2(out, [{"note": 60, "frames": _sine(60)}], 44100, name="Nexus")
|
|
with caplog.at_level(logging.WARNING):
|
|
with open(out, "rb") as f:
|
|
sf2 = sf2utils.Sf2File(f)
|
|
assert len([s for s in sf2.samples if s.end > s.start]) == 1
|
|
assert not any("corrupted" in r.message for r in caplog.records)
|
|
|
|
def test_autosample_vst2_native_bridge(tmp_path):
|
|
"""Nhanh VST2: render qua native bridge (fake_vst2.dll) → SF2 hop le,
|
|
khong can pedalboard. Skip tren nen khong phai Windows / thieu DLL."""
|
|
if sys.platform.startswith("win") and os.path.isfile(FAKE_VST2):
|
|
from tools.autosample_vsti import autosample_sf2
|
|
out = str(tmp_path / "vst2.sf2")
|
|
res = autosample_sf2("fake_vst2", out, low=60, high=64, step=2,
|
|
duration=0.3, release=0.2,
|
|
extra_vst_dirs=[os.path.dirname(FAKE_VST2)])
|
|
assert res["note_count"] == 3
|
|
assert res["size_bytes"] == os.path.getsize(out)
|
|
with open(out, "rb") as f:
|
|
sf2 = sf2utils.Sf2File(f)
|
|
assert len([s for s in sf2.samples if s.end > s.start]) == 3
|
|
else:
|
|
pytest.skip("fake_vst2.dll chi chay Windows (bridge ctypes WinDLL)")
|