mirror of
https://github.com/firestar5683/StarPilot.git
synced 2026-10-04 13:24:13 +08:00
firestar5683
40 lines
1.5 KiB
Python
40 lines
1.5 KiB
Python
import sys
|
|
import tempfile
|
|
import unittest
|
|
from pathlib import Path
|
|
import numpy as np
|
|
from scipy.io import wavfile
|
|
sys.path.insert(0, str(Path(__file__).resolve().parents[1] / 'tools'))
|
|
from review_hook_audio import screen, assembly_check
|
|
|
|
|
|
class AudioScreeningTests(unittest.TestCase):
|
|
def test_expected_crossfade_is_excluded_but_other_changes_are_reported(self):
|
|
parts = [np.zeros((40, 2)), np.ones((40, 2))]
|
|
core = np.concatenate(parts)
|
|
core[20:40] = .3
|
|
self.assertEqual(assembly_check(core, parts, 10)['status'], 'matches outside crossfades')
|
|
core[2] = .1
|
|
self.assertEqual(assembly_check(core, parts, 10)['status'], 'unexpected difference outside crossfades')
|
|
|
|
def test_detects_clipping_silence_and_repeated_blocks(self):
|
|
rate = 48000
|
|
block = np.random.default_rng(17).normal(0, .05, (rate, 2)).astype('float32')
|
|
audio = np.concatenate([block, block, np.zeros((rate, 2), 'float32')])
|
|
audio[0, 0] = 1.01
|
|
with tempfile.TemporaryDirectory() as folder:
|
|
path = Path(folder) / 'test.wav'
|
|
wavfile.write(path, rate, audio)
|
|
result = screen(path)
|
|
self.assertEqual(result['samples_at_or_above_full_scale'], 1)
|
|
self.assertAlmostEqual(result['longest_quiet_seconds'], 1)
|
|
self.assertEqual(result['exact_repeated_nonsilent_1s_blocks'], [])
|
|
audio[0] = block[0]
|
|
wavfile.write(path, rate, audio)
|
|
result = screen(path)
|
|
self.assertEqual(result['exact_repeated_nonsilent_1s_blocks'], [[0, 1]])
|
|
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|