385 lines
14 KiB
Python
385 lines
14 KiB
Python
from test.support import findfile, run_unittest, TESTFN, unlink
|
|
import unittest
|
|
import os
|
|
import io
|
|
import struct
|
|
import pickle
|
|
|
|
import aifc
|
|
|
|
|
|
class AIFCTest(unittest.TestCase):
|
|
|
|
def setUp(self):
|
|
self.f = self.fout = None
|
|
self.sndfilepath = findfile('Sine-1000Hz-300ms.aif')
|
|
|
|
def tearDown(self):
|
|
if self.f is not None:
|
|
self.f.close()
|
|
if self.fout is not None:
|
|
try:
|
|
self.fout.close()
|
|
except (aifc.Error, AttributeError):
|
|
pass
|
|
unlink(TESTFN)
|
|
unlink(TESTFN + '.aiff')
|
|
|
|
def test_skipunknown(self):
|
|
#Issue 2245
|
|
#This file contains chunk types aifc doesn't recognize.
|
|
self.f = aifc.open(self.sndfilepath)
|
|
|
|
def test_params(self):
|
|
f = self.f = aifc.open(self.sndfilepath)
|
|
params = f.getparams()
|
|
self.assertEqual(f.getfp().name, self.sndfilepath)
|
|
self.assertEqual(f.getnchannels(), 2)
|
|
self.assertEqual(f.getsampwidth(), 2)
|
|
self.assertEqual(f.getframerate(), 48000)
|
|
self.assertEqual(f.getnframes(), 14400)
|
|
self.assertEqual(f.getcomptype(), b'NONE')
|
|
self.assertEqual(f.getcompname(), b'not compressed')
|
|
self.assertEqual(
|
|
f.getparams(),
|
|
(2, 2, 48000, 14400, b'NONE', b'not compressed'),
|
|
)
|
|
|
|
params = f.getparams()
|
|
self.assertEqual(params.nchannels, 2)
|
|
self.assertEqual(params.sampwidth, 2)
|
|
self.assertEqual(params.framerate, 48000)
|
|
self.assertEqual(params.nframes, 14400)
|
|
self.assertEqual(params.comptype, b'NONE')
|
|
self.assertEqual(params.compname, b'not compressed')
|
|
|
|
def test_params_added(self):
|
|
f = self.f = aifc.open(TESTFN, 'wb')
|
|
f.aiff()
|
|
f.setparams((1, 1, 1, 1, b'NONE', b''))
|
|
f.close()
|
|
|
|
f = self.f = aifc.open(TESTFN, 'rb')
|
|
params = f.getparams()
|
|
self.assertEqual(params.nchannels, f.getnchannels())
|
|
self.assertEqual(params.sampwidth, f.getsampwidth())
|
|
self.assertEqual(params.framerate, f.getframerate())
|
|
self.assertEqual(params.nframes, f.getnframes())
|
|
self.assertEqual(params.comptype, f.getcomptype())
|
|
self.assertEqual(params.compname, f.getcompname())
|
|
|
|
def test_getparams_picklable(self):
|
|
self.f = aifc.open(self.sndfilepath)
|
|
params = self.f.getparams()
|
|
dump = pickle.dumps(params)
|
|
self.assertEqual(pickle.loads(dump), params)
|
|
self.f.close()
|
|
|
|
def test_context_manager(self):
|
|
with open(self.sndfilepath, 'rb') as testfile:
|
|
with aifc.open(testfile) as f:
|
|
pass
|
|
self.assertEqual(testfile.closed, True)
|
|
with open(TESTFN, 'wb') as testfile:
|
|
with self.assertRaises(aifc.Error):
|
|
with aifc.open(testfile, 'wb') as fout:
|
|
pass
|
|
self.assertEqual(testfile.closed, True)
|
|
fout.close() # do nothing
|
|
|
|
def test_read(self):
|
|
f = self.f = aifc.open(self.sndfilepath)
|
|
self.assertEqual(f.readframes(0), b'')
|
|
self.assertEqual(f.tell(), 0)
|
|
self.assertEqual(f.readframes(2), b'\x00\x00\x00\x00\x0b\xd4\x0b\xd4')
|
|
f.rewind()
|
|
pos0 = f.tell()
|
|
self.assertEqual(pos0, 0)
|
|
self.assertEqual(f.readframes(2), b'\x00\x00\x00\x00\x0b\xd4\x0b\xd4')
|
|
pos2 = f.tell()
|
|
self.assertEqual(pos2, 2)
|
|
self.assertEqual(f.readframes(2), b'\x17t\x17t"\xad"\xad')
|
|
f.setpos(pos2)
|
|
self.assertEqual(f.readframes(2), b'\x17t\x17t"\xad"\xad')
|
|
f.setpos(pos0)
|
|
self.assertEqual(f.readframes(2), b'\x00\x00\x00\x00\x0b\xd4\x0b\xd4')
|
|
with self.assertRaises(aifc.Error):
|
|
f.setpos(-1)
|
|
with self.assertRaises(aifc.Error):
|
|
f.setpos(f.getnframes() + 1)
|
|
|
|
def test_write(self):
|
|
f = self.f = aifc.open(self.sndfilepath)
|
|
fout = self.fout = aifc.open(TESTFN, 'wb')
|
|
fout.aifc()
|
|
fout.setparams(f.getparams())
|
|
for frame in range(f.getnframes()):
|
|
fout.writeframes(f.readframes(1))
|
|
fout.close()
|
|
fout = self.fout = aifc.open(TESTFN, 'rb')
|
|
f.rewind()
|
|
self.assertEqual(f.getparams(), fout.getparams())
|
|
self.assertEqual(f.readframes(5), fout.readframes(5))
|
|
|
|
def test_compress(self):
|
|
f = self.f = aifc.open(self.sndfilepath)
|
|
fout = self.fout = aifc.open(TESTFN, 'wb')
|
|
fout.aifc()
|
|
fout.setnchannels(f.getnchannels())
|
|
fout.setsampwidth(f.getsampwidth())
|
|
fout.setframerate(f.getframerate())
|
|
fout.setcomptype(b'ULAW', b'foo')
|
|
for frame in range(f.getnframes()):
|
|
fout.writeframes(f.readframes(1))
|
|
fout.close()
|
|
self.assertLess(
|
|
os.stat(TESTFN).st_size,
|
|
os.stat(self.sndfilepath).st_size*0.75,
|
|
)
|
|
fout = self.fout = aifc.open(TESTFN, 'rb')
|
|
f.rewind()
|
|
self.assertEqual(f.getparams()[0:3], fout.getparams()[0:3])
|
|
self.assertEqual(fout.getcomptype(), b'ULAW')
|
|
self.assertEqual(fout.getcompname(), b'foo')
|
|
|
|
def test_close(self):
|
|
class Wrapfile(object):
|
|
def __init__(self, file):
|
|
self.file = open(file, 'rb')
|
|
self.closed = False
|
|
def close(self):
|
|
self.file.close()
|
|
self.closed = True
|
|
def __getattr__(self, attr): return getattr(self.file, attr)
|
|
testfile = Wrapfile(self.sndfilepath)
|
|
f = self.f = aifc.open(testfile)
|
|
self.assertEqual(testfile.closed, False)
|
|
f.close()
|
|
self.assertEqual(testfile.closed, True)
|
|
testfile = open(TESTFN, 'wb')
|
|
fout = aifc.open(testfile, 'wb')
|
|
self.assertFalse(testfile.closed)
|
|
with self.assertRaises(aifc.Error):
|
|
fout.close()
|
|
self.assertTrue(testfile.closed)
|
|
fout.close() # do nothing
|
|
|
|
def test_write_header_comptype_sampwidth(self):
|
|
for comptype in (b'ULAW', b'ulaw', b'ALAW', b'alaw', b'G722'):
|
|
fout = aifc.open(io.BytesIO(), 'wb')
|
|
fout.setnchannels(1)
|
|
fout.setframerate(1)
|
|
fout.setcomptype(comptype, b'')
|
|
fout.close()
|
|
self.assertEqual(fout.getsampwidth(), 2)
|
|
fout.initfp(None)
|
|
|
|
def test_write_markers_values(self):
|
|
fout = aifc.open(io.BytesIO(), 'wb')
|
|
self.assertEqual(fout.getmarkers(), None)
|
|
fout.setmark(1, 0, b'foo1')
|
|
fout.setmark(1, 1, b'foo2')
|
|
self.assertEqual(fout.getmark(1), (1, 1, b'foo2'))
|
|
self.assertEqual(fout.getmarkers(), [(1, 1, b'foo2')])
|
|
fout.initfp(None)
|
|
|
|
def test_read_markers(self):
|
|
fout = self.fout = aifc.open(TESTFN, 'wb')
|
|
fout.aiff()
|
|
fout.setparams((1, 1, 1, 1, b'NONE', b''))
|
|
fout.setmark(1, 0, b'odd')
|
|
fout.setmark(2, 0, b'even')
|
|
fout.writeframes(b'\x00')
|
|
fout.close()
|
|
f = self.f = aifc.open(TESTFN, 'rb')
|
|
self.assertEqual(f.getmarkers(), [(1, 0, b'odd'), (2, 0, b'even')])
|
|
self.assertEqual(f.getmark(1), (1, 0, b'odd'))
|
|
self.assertEqual(f.getmark(2), (2, 0, b'even'))
|
|
self.assertRaises(aifc.Error, f.getmark, 3)
|
|
|
|
|
|
class AIFCLowLevelTest(unittest.TestCase):
|
|
|
|
def test_read_written(self):
|
|
def read_written(self, what):
|
|
f = io.BytesIO()
|
|
getattr(aifc, '_write_' + what)(f, x)
|
|
f.seek(0)
|
|
return getattr(aifc, '_read_' + what)(f)
|
|
for x in (-1, 0, 0.1, 1):
|
|
self.assertEqual(read_written(x, 'float'), x)
|
|
for x in (float('NaN'), float('Inf')):
|
|
self.assertEqual(read_written(x, 'float'), aifc._HUGE_VAL)
|
|
for x in (b'', b'foo', b'a' * 255):
|
|
self.assertEqual(read_written(x, 'string'), x)
|
|
for x in (-0x7FFFFFFF, -1, 0, 1, 0x7FFFFFFF):
|
|
self.assertEqual(read_written(x, 'long'), x)
|
|
for x in (0, 1, 0xFFFFFFFF):
|
|
self.assertEqual(read_written(x, 'ulong'), x)
|
|
for x in (-0x7FFF, -1, 0, 1, 0x7FFF):
|
|
self.assertEqual(read_written(x, 'short'), x)
|
|
for x in (0, 1, 0xFFFF):
|
|
self.assertEqual(read_written(x, 'ushort'), x)
|
|
|
|
def test_read_raises(self):
|
|
f = io.BytesIO(b'\x00')
|
|
self.assertRaises(EOFError, aifc._read_ulong, f)
|
|
self.assertRaises(EOFError, aifc._read_long, f)
|
|
self.assertRaises(EOFError, aifc._read_ushort, f)
|
|
self.assertRaises(EOFError, aifc._read_short, f)
|
|
|
|
def test_write_long_string_raises(self):
|
|
f = io.BytesIO()
|
|
with self.assertRaises(ValueError):
|
|
aifc._write_string(f, b'too long' * 255)
|
|
|
|
def test_wrong_open_mode(self):
|
|
with self.assertRaises(aifc.Error):
|
|
aifc.open(TESTFN, 'wrong_mode')
|
|
|
|
def test_read_wrong_form(self):
|
|
b1 = io.BytesIO(b'WRNG' + struct.pack('>L', 0))
|
|
b2 = io.BytesIO(b'FORM' + struct.pack('>L', 4) + b'WRNG')
|
|
self.assertRaises(aifc.Error, aifc.open, b1)
|
|
self.assertRaises(aifc.Error, aifc.open, b2)
|
|
|
|
def test_read_no_comm_chunk(self):
|
|
b = io.BytesIO(b'FORM' + struct.pack('>L', 4) + b'AIFF')
|
|
self.assertRaises(aifc.Error, aifc.open, b)
|
|
|
|
def test_read_wrong_compression_type(self):
|
|
b = b'FORM' + struct.pack('>L', 4) + b'AIFC'
|
|
b += b'COMM' + struct.pack('>LhlhhLL', 23, 0, 0, 0, 0, 0, 0)
|
|
b += b'WRNG' + struct.pack('B', 0)
|
|
self.assertRaises(aifc.Error, aifc.open, io.BytesIO(b))
|
|
|
|
def test_read_wrong_marks(self):
|
|
b = b'FORM' + struct.pack('>L', 4) + b'AIFF'
|
|
b += b'COMM' + struct.pack('>LhlhhLL', 18, 0, 0, 0, 0, 0, 0)
|
|
b += b'SSND' + struct.pack('>L', 8) + b'\x00' * 8
|
|
b += b'MARK' + struct.pack('>LhB', 3, 1, 1)
|
|
with self.assertWarns(UserWarning):
|
|
f = aifc.open(io.BytesIO(b))
|
|
self.assertEqual(f.getmarkers(), None)
|
|
|
|
def test_read_comm_kludge_compname_even(self):
|
|
b = b'FORM' + struct.pack('>L', 4) + b'AIFC'
|
|
b += b'COMM' + struct.pack('>LhlhhLL', 18, 0, 0, 0, 0, 0, 0)
|
|
b += b'NONE' + struct.pack('B', 4) + b'even' + b'\x00'
|
|
b += b'SSND' + struct.pack('>L', 8) + b'\x00' * 8
|
|
with self.assertWarns(UserWarning):
|
|
f = aifc.open(io.BytesIO(b))
|
|
self.assertEqual(f.getcompname(), b'even')
|
|
|
|
def test_read_comm_kludge_compname_odd(self):
|
|
b = b'FORM' + struct.pack('>L', 4) + b'AIFC'
|
|
b += b'COMM' + struct.pack('>LhlhhLL', 18, 0, 0, 0, 0, 0, 0)
|
|
b += b'NONE' + struct.pack('B', 3) + b'odd'
|
|
b += b'SSND' + struct.pack('>L', 8) + b'\x00' * 8
|
|
with self.assertWarns(UserWarning):
|
|
f = aifc.open(io.BytesIO(b))
|
|
self.assertEqual(f.getcompname(), b'odd')
|
|
|
|
def test_write_params_raises(self):
|
|
fout = aifc.open(io.BytesIO(), 'wb')
|
|
wrong_params = (0, 0, 0, 0, b'WRNG', '')
|
|
self.assertRaises(aifc.Error, fout.setparams, wrong_params)
|
|
self.assertRaises(aifc.Error, fout.getparams)
|
|
self.assertRaises(aifc.Error, fout.setnchannels, 0)
|
|
self.assertRaises(aifc.Error, fout.getnchannels)
|
|
self.assertRaises(aifc.Error, fout.setsampwidth, 0)
|
|
self.assertRaises(aifc.Error, fout.getsampwidth)
|
|
self.assertRaises(aifc.Error, fout.setframerate, 0)
|
|
self.assertRaises(aifc.Error, fout.getframerate)
|
|
self.assertRaises(aifc.Error, fout.setcomptype, b'WRNG', '')
|
|
fout.aiff()
|
|
fout.setnchannels(1)
|
|
fout.setsampwidth(1)
|
|
fout.setframerate(1)
|
|
fout.setnframes(1)
|
|
fout.writeframes(b'\x00')
|
|
self.assertRaises(aifc.Error, fout.setparams, (1, 1, 1, 1, 1, 1))
|
|
self.assertRaises(aifc.Error, fout.setnchannels, 1)
|
|
self.assertRaises(aifc.Error, fout.setsampwidth, 1)
|
|
self.assertRaises(aifc.Error, fout.setframerate, 1)
|
|
self.assertRaises(aifc.Error, fout.setnframes, 1)
|
|
self.assertRaises(aifc.Error, fout.setcomptype, b'NONE', '')
|
|
self.assertRaises(aifc.Error, fout.aiff)
|
|
self.assertRaises(aifc.Error, fout.aifc)
|
|
|
|
def test_write_params_singles(self):
|
|
fout = aifc.open(io.BytesIO(), 'wb')
|
|
fout.aifc()
|
|
fout.setnchannels(1)
|
|
fout.setsampwidth(2)
|
|
fout.setframerate(3)
|
|
fout.setnframes(4)
|
|
fout.setcomptype(b'NONE', b'name')
|
|
self.assertEqual(fout.getnchannels(), 1)
|
|
self.assertEqual(fout.getsampwidth(), 2)
|
|
self.assertEqual(fout.getframerate(), 3)
|
|
self.assertEqual(fout.getnframes(), 0)
|
|
self.assertEqual(fout.tell(), 0)
|
|
self.assertEqual(fout.getcomptype(), b'NONE')
|
|
self.assertEqual(fout.getcompname(), b'name')
|
|
fout.writeframes(b'\x00' * 4 * fout.getsampwidth() * fout.getnchannels())
|
|
self.assertEqual(fout.getnframes(), 4)
|
|
self.assertEqual(fout.tell(), 4)
|
|
|
|
def test_write_params_bunch(self):
|
|
fout = aifc.open(io.BytesIO(), 'wb')
|
|
fout.aifc()
|
|
p = (1, 2, 3, 4, b'NONE', b'name')
|
|
fout.setparams(p)
|
|
self.assertEqual(fout.getparams(), p)
|
|
fout.initfp(None)
|
|
|
|
def test_write_header_raises(self):
|
|
fout = aifc.open(io.BytesIO(), 'wb')
|
|
self.assertRaises(aifc.Error, fout.close)
|
|
fout = aifc.open(io.BytesIO(), 'wb')
|
|
fout.setnchannels(1)
|
|
self.assertRaises(aifc.Error, fout.close)
|
|
fout = aifc.open(io.BytesIO(), 'wb')
|
|
fout.setnchannels(1)
|
|
fout.setsampwidth(1)
|
|
self.assertRaises(aifc.Error, fout.close)
|
|
|
|
def test_write_header_comptype_raises(self):
|
|
for comptype in (b'ULAW', b'ulaw', b'ALAW', b'alaw', b'G722'):
|
|
fout = aifc.open(io.BytesIO(), 'wb')
|
|
fout.setsampwidth(1)
|
|
fout.setcomptype(comptype, b'')
|
|
self.assertRaises(aifc.Error, fout.close)
|
|
fout.initfp(None)
|
|
|
|
def test_write_markers_raises(self):
|
|
fout = aifc.open(io.BytesIO(), 'wb')
|
|
self.assertRaises(aifc.Error, fout.setmark, 0, 0, b'')
|
|
self.assertRaises(aifc.Error, fout.setmark, 1, -1, b'')
|
|
self.assertRaises(aifc.Error, fout.setmark, 1, 0, None)
|
|
self.assertRaises(aifc.Error, fout.getmark, 1)
|
|
fout.initfp(None)
|
|
|
|
def test_write_aiff_by_extension(self):
|
|
sampwidth = 2
|
|
filename = TESTFN + '.aiff'
|
|
fout = self.fout = aifc.open(filename, 'wb')
|
|
self.addCleanup(unlink, filename)
|
|
fout.setparams((1, sampwidth, 1, 1, b'ULAW', b''))
|
|
frames = b'\x00' * fout.getnchannels() * sampwidth
|
|
fout.writeframes(frames)
|
|
fout.close()
|
|
f = self.f = aifc.open(filename, 'rb')
|
|
self.assertEqual(f.getcomptype(), b'NONE')
|
|
f.close()
|
|
|
|
|
|
def test_main():
|
|
run_unittest(AIFCTest)
|
|
run_unittest(AIFCLowLevelTest)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|