mirror of
https://github.com/iluvcapra/wavinfo.git
synced 2026-10-02 01:37:22 +00:00
75 lines
2.6 KiB
Python
75 lines
2.6 KiB
Python
from glob import glob
|
|
from unittest import TestCase
|
|
|
|
import wavinfo
|
|
|
|
|
|
class TestCue(TestCase):
|
|
def setUp(self) -> None:
|
|
self.test_files = glob("tests/test_files/cue_chunks/*.wav")
|
|
return super().setUp()
|
|
|
|
def test_enumerate(self):
|
|
file1 = "tests/test_files/cue_chunks/STE-000.wav"
|
|
w1 = wavinfo.WavInfoReader(file1)
|
|
self.assertIsNotNone(w1.cues)
|
|
assert w1.cues is not None
|
|
vals = list(w1.cues.each_cue())
|
|
self.assertEqual(vals, [(1, 29616), (2, 74592), (3, 121200)])
|
|
|
|
def test_labels_notes(self):
|
|
file = "tests/test_files/cue_chunks/izotoperx_cues_test.wav"
|
|
w1 = wavinfo.WavInfoReader(file)
|
|
self.assertIsNotNone(w1.cues)
|
|
assert w1.cues is not None
|
|
|
|
for name, _ in w1.cues.each_cue():
|
|
self.assertIn(name, [1, 2, 3])
|
|
label, note = w1.cues.label_and_note(name)
|
|
if name == 1:
|
|
self.assertEqual("Marker 1", label)
|
|
self.assertIsNone(note)
|
|
|
|
def test_range(self):
|
|
file = "tests/test_files/cue_chunks/izotoperx_cues_test.wav"
|
|
w1 = wavinfo.WavInfoReader(file)
|
|
self.assertIsNotNone(w1.cues)
|
|
assert w1.cues is not None
|
|
|
|
self.assertEqual(w1.cues.range(3), 10000)
|
|
|
|
def test_encoding_fallback(self):
|
|
"""
|
|
Added this after I noticed that iZotope RX seems to just encode "notes"
|
|
as utf-8 without bothering to dump this info into the ltxt or
|
|
specifying an encoding by some other means.
|
|
"""
|
|
file = "tests/test_files/cue_chunks/izotoperx_cues_test.wav"
|
|
w = wavinfo.WavInfoReader(file, info_encoding="utf-8")
|
|
expected = (
|
|
"Лорем ипсум долор сит амет, тимеам вивендум хас ет, "
|
|
"цу адолесценс дефинитионес еам."
|
|
)
|
|
|
|
assert w.cues is not None
|
|
note = [n for n in w.cues.notes if n.name == 3]
|
|
self.assertEqual(len(note), 1)
|
|
self.assertEqual(note[0].text, expected)
|
|
|
|
def test_label(self):
|
|
file = "tests/test_files/cue_chunks/izotoperx_cues_test.wav"
|
|
w = wavinfo.WavInfoReader(file)
|
|
|
|
self.assertIsNotNone(w.cues)
|
|
assert w.cues is not None
|
|
|
|
self.assertEqual(len(w.cues.labels), 3)
|
|
for label in w.cues.labels:
|
|
self.assertIn(label.name, [1, 2, 3])
|
|
if label.name == 1:
|
|
self.assertEqual(label.text, "Marker 1")
|
|
elif label.name == 2:
|
|
self.assertEqual(label.text, "Marker 2")
|
|
elif label.name == 3:
|
|
self.assertEqual(label.text, "Marker 3")
|