119 lines
3.9 KiB
Python
119 lines
3.9 KiB
Python
from test.support import findfile, run_unittest, TESTFN
|
|
import unittest
|
|
import os
|
|
|
|
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
|
|
try:
|
|
os.remove(TESTFN)
|
|
except OSError:
|
|
pass
|
|
|
|
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)
|
|
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'),
|
|
)
|
|
|
|
def test_read(self):
|
|
f = self.f = aifc.open(self.sndfilepath)
|
|
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')
|
|
|
|
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')
|
|
# XXX: this test fails, not sure if it should succeed or not
|
|
# self.assertEqual(f.readframes(5), fout.readframes(5))
|
|
|
|
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)
|
|
|
|
|
|
def test_main():
|
|
run_unittest(AIFCTest)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|