1 from test.test_support import (findfile, TESTFN, unlink, captured_stdout, 2 run_unittest) 3 import unittest 4 from test import audiotests 5 import os 6 import io 7 import sys 8 import struct 9 import aifc 10 11 12 class AifcTest(audiotests.AudioWriteTests, 13 audiotests.AudioTestsWithSourceFile): 14 module = aifc 15 close_fd = True 16 test_unseekable_read = None 17 18 19 class AifcPCM8Test(AifcTest, unittest.TestCase): 20 sndfilename = 'pluck-pcm8.aiff' 21 sndfilenframes = 3307 22 nchannels = 2 23 sampwidth = 1 24 framerate = 11025 25 nframes = 48 26 comptype = 'NONE' 27 compname = 'not compressed' 28 frames = audiotests.fromhex("""\ 29 02FF 4B00 3104 8008 CB06 4803 BF01 03FE B8FA B4F3 29EB 1AE6 \ 30 EDE4 C6E2 0EE0 EFE0 57E2 FBE8 13EF D8F7 97FB F5FC 08FB DFFB \ 31 11FA 3EFB BCFC 66FF CF04 4309 C10E 5112 EE17 8216 7F14 8012 \ 32 490E 520D EF0F CE0F E40C 630A 080A 2B0B 510E 8B11 B60E 440A \ 33 """) 34 35 36 class AifcPCM16Test(AifcTest, unittest.TestCase): 37 sndfilename = 'pluck-pcm16.aiff' 38 sndfilenframes = 3307 39 nchannels = 2 40 sampwidth = 2 41 framerate = 11025 42 nframes = 48 43 comptype = 'NONE' 44 compname = 'not compressed' 45 frames = audiotests.fromhex("""\ 46 022EFFEA 4B5D00F6 311804EA 80E10840 CBE106B1 48A903F5 BFE601B2 036CFE7B \ 47 B858FA3E B4B1F34F 299AEBCA 1A5DE6DA EDFAE491 C628E275 0E09E0B5 EF2AE029 \ 48 5758E271 FB35E83F 1376EF86 D82BF727 9790FB76 F5FAFC0F 0867FB9C DF30FB43 \ 49 117EFA36 3EE5FB5B BC79FCB1 66D9FF5D CF150412 431D097C C1BA0EC8 512112A1 \ 50 EEE21753 82071665 7FFF1443 8004128F 49A20EAF 52BB0DBA EFB40F60 CE3C0FBF \ 51 E4B30CEC 63430A5C 08C80A20 2BBB0B08 514A0E43 8BCF1139 B6F60EEB 44120A5E \ 52 """) 53 54 55 class AifcPCM24Test(AifcTest, unittest.TestCase): 56 sndfilename = 'pluck-pcm24.aiff' 57 sndfilenframes = 3307 58 nchannels = 2 59 sampwidth = 3 60 framerate = 11025 61 nframes = 48 62 comptype = 'NONE' 63 compname = 'not compressed' 64 frames = audiotests.fromhex("""\ 65 022D65FFEB9D 4B5A0F00FA54 3113C304EE2B 80DCD6084303 \ 66 CBDEC006B261 48A99803F2F8 BFE82401B07D 036BFBFE7B5D \ 67 B85756FA3EC9 B4B055F3502B 299830EBCB62 1A5CA7E6D99A \ 68 EDFA3EE491BD C625EBE27884 0E05A9E0B6CF EF2929E02922 \ 69 5758D8E27067 FB3557E83E16 1377BFEF8402 D82C5BF7272A \ 70 978F16FB7745 F5F865FC1013 086635FB9C4E DF30FCFB40EE \ 71 117FE0FA3438 3EE6B8FB5AC3 BC77A3FCB2F4 66D6DAFF5F32 \ 72 CF13B9041275 431D69097A8C C1BB600EC74E 5120B912A2BA \ 73 EEDF641754C0 8207001664B7 7FFFFF14453F 8000001294E6 \ 74 499C1B0EB3B2 52B73E0DBCA0 EFB2B20F5FD8 CE3CDB0FBE12 \ 75 E4B49C0CEA2D 6344A80A5A7C 08C8FE0A1FFE 2BB9860B0A0E \ 76 51486F0E44E1 8BCC64113B05 B6F4EC0EEB36 4413170A5B48 \ 77 """) 78 79 80 class AifcPCM32Test(AifcTest, unittest.TestCase): 81 sndfilename = 'pluck-pcm32.aiff' 82 sndfilenframes = 3307 83 nchannels = 2 84 sampwidth = 4 85 framerate = 11025 86 nframes = 48 87 comptype = 'NONE' 88 compname = 'not compressed' 89 frames = audiotests.fromhex("""\ 90 022D65BCFFEB9D92 4B5A0F8000FA549C 3113C34004EE2BC0 80DCD680084303E0 \ 91 CBDEC0C006B26140 48A9980003F2F8FC BFE8248001B07D92 036BFB60FE7B5D34 \ 92 B8575600FA3EC920 B4B05500F3502BC0 29983000EBCB6240 1A5CA7A0E6D99A60 \ 93 EDFA3E80E491BD40 C625EB80E27884A0 0E05A9A0E0B6CFE0 EF292940E0292280 \ 94 5758D800E2706700 FB3557D8E83E1640 1377BF00EF840280 D82C5B80F7272A80 \ 95 978F1600FB774560 F5F86510FC101364 086635A0FB9C4E20 DF30FC40FB40EE28 \ 96 117FE0A0FA3438B0 3EE6B840FB5AC3F0 BC77A380FCB2F454 66D6DA80FF5F32B4 \ 97 CF13B980041275B0 431D6980097A8C00 C1BB60000EC74E00 5120B98012A2BAA0 \ 98 EEDF64C01754C060 820700001664B780 7FFFFFFF14453F40 800000001294E6E0 \ 99 499C1B000EB3B270 52B73E000DBCA020 EFB2B2E00F5FD880 CE3CDB400FBE1270 \ 100 E4B49CC00CEA2D90 6344A8800A5A7CA0 08C8FE800A1FFEE0 2BB986C00B0A0E00 \ 101 51486F800E44E190 8BCC6480113B0580 B6F4EC000EEB3630 441317800A5B48A0 \ 102 """) 103 104 105 class AifcULAWTest(AifcTest, unittest.TestCase): 106 sndfilename = 'pluck-ulaw.aifc' 107 sndfilenframes = 3307 108 nchannels = 2 109 sampwidth = 2 110 framerate = 11025 111 nframes = 48 112 comptype = 'ulaw' 113 compname = '' 114 frames = audiotests.fromhex("""\ 115 022CFFE8 497C0104 307C04DC 8284083C CB84069C 497C03DC BE8401AC 036CFE74 \ 116 B684FA24 B684F344 2A7CEC04 19FCE704 EE04E504 C584E204 0E3CE104 EF04DF84 \ 117 557CE204 FB24E804 12FCEF04 D784F744 9684FB64 F5C4FC24 083CFBA4 DF84FB24 \ 118 11FCFA24 3E7CFB64 BA84FCB4 657CFF5C CF84041C 417C093C C1840EBC 517C12FC \ 119 EF0416FC 828415FC 7D7C13FC 828412FC 497C0EBC 517C0DBC F0040F3C CD840FFC \ 120 E5040CBC 617C0A3C 08BC0A3C 2C7C0B3C 517C0E3C 8A8410FC B6840EBC 457C0A3C \ 121 """) 122 if sys.byteorder != 'big': 123 frames = audiotests.byteswap2(frames) 124 125 126 class AifcMiscTest(audiotests.AudioTests, unittest.TestCase): 127 def test_skipunknown(self): 128 #Issue 2245 129 #This file contains chunk types aifc doesn't recognize. 130 self.f = aifc.open(findfile('Sine-1000Hz-300ms.aif')) 131 132 def test_close_opened_files_on_error(self): 133 non_aifc_file = findfile('pluck-pcm8.wav', subdir='audiodata') 134 135 class Aifc(aifc.Aifc_read): 136 def __init__(self): 137 pass 138 139 a = Aifc() 140 with self.assertRaises(aifc.Error): 141 aifc.Aifc_read.__init__(a, non_aifc_file) 142 self.assertTrue(a._file.closed) 143 144 def test_write_markers_values(self): 145 fout = aifc.open(io.BytesIO(), 'wb') 146 self.assertEqual(fout.getmarkers(), None) 147 fout.setmark(1, 0, 'foo1') 148 fout.setmark(1, 1, 'foo2') 149 self.assertEqual(fout.getmark(1), (1, 1, 'foo2')) 150 self.assertEqual(fout.getmarkers(), [(1, 1, 'foo2')]) 151 fout.initfp(None) 152 153 def test_read_markers(self): 154 fout = self.fout = aifc.open(TESTFN, 'wb') 155 fout.aiff() 156 fout.setparams((1, 1, 1, 1, 'NONE', '')) 157 fout.setmark(1, 0, 'odd') 158 fout.setmark(2, 0, 'even') 159 fout.writeframes('\x00') 160 fout.close() 161 f = self.f = aifc.open(TESTFN, 'rb') 162 self.assertEqual(f.getmarkers(), [(1, 0, 'odd'), (2, 0, 'even')]) 163 self.assertEqual(f.getmark(1), (1, 0, 'odd')) 164 self.assertEqual(f.getmark(2), (2, 0, 'even')) 165 self.assertRaises(aifc.Error, f.getmark, 3) 166 167 168 class AIFCLowLevelTest(unittest.TestCase): 169 170 def test_read_written(self): 171 def read_written(self, what): 172 f = io.BytesIO() 173 getattr(aifc, '_write_' + what)(f, x) 174 f.seek(0) 175 return getattr(aifc, '_read_' + what)(f) 176 for x in (-1, 0, 0.1, 1): 177 self.assertEqual(read_written(x, 'float'), x) 178 for x in (float('NaN'), float('Inf')): 179 self.assertEqual(read_written(x, 'float'), aifc._HUGE_VAL) 180 for x in ('', 'foo', 'a' * 255): 181 self.assertEqual(read_written(x, 'string'), x) 182 for x in (-0x7FFFFFFF, -1, 0, 1, 0x7FFFFFFF): 183 self.assertEqual(read_written(x, 'long'), x) 184 for x in (0, 1, 0xFFFFFFFF): 185 self.assertEqual(read_written(x, 'ulong'), x) 186 for x in (-0x7FFF, -1, 0, 1, 0x7FFF): 187 self.assertEqual(read_written(x, 'short'), x) 188 for x in (0, 1, 0xFFFF): 189 self.assertEqual(read_written(x, 'ushort'), x) 190 191 def test_read_raises(self): 192 f = io.BytesIO('\x00') 193 self.assertRaises(EOFError, aifc._read_ulong, f) 194 self.assertRaises(EOFError, aifc._read_long, f) 195 self.assertRaises(EOFError, aifc._read_ushort, f) 196 self.assertRaises(EOFError, aifc._read_short, f) 197 198 def test_write_long_string_raises(self): 199 f = io.BytesIO() 200 with self.assertRaises(ValueError): 201 aifc._write_string(f, 'too long' * 255) 202 203 def test_wrong_open_mode(self): 204 with self.assertRaises(aifc.Error): 205 aifc.open(TESTFN, 'wrong_mode') 206 207 def test_read_wrong_form(self): 208 b1 = io.BytesIO('WRNG' + struct.pack('>L', 0)) 209 b2 = io.BytesIO('FORM' + struct.pack('>L', 4) + 'WRNG') 210 self.assertRaises(aifc.Error, aifc.open, b1) 211 self.assertRaises(aifc.Error, aifc.open, b2) 212 213 def test_read_no_comm_chunk(self): 214 b = io.BytesIO('FORM' + struct.pack('>L', 4) + 'AIFF') 215 self.assertRaises(aifc.Error, aifc.open, b) 216 217 def test_read_no_ssnd_chunk(self): 218 b = b'FORM' + struct.pack('>L', 4) + b'AIFC' 219 b += b'COMM' + struct.pack('>LhlhhLL', 38, 0, 0, 0, 0, 0, 0) 220 b += b'NONE' + struct.pack('B', 14) + b'not compressed' + b'\x00' 221 with self.assertRaisesRegexp(aifc.Error, 'COMM chunk and/or SSND chunk' 222 ' missing'): 223 aifc.open(io.BytesIO(b)) 224 225 def test_read_wrong_compression_type(self): 226 b = 'FORM' + struct.pack('>L', 4) + 'AIFC' 227 b += 'COMM' + struct.pack('>LhlhhLL', 23, 0, 0, 0, 0, 0, 0) 228 b += 'WRNG' + struct.pack('B', 0) 229 self.assertRaises(aifc.Error, aifc.open, io.BytesIO(b)) 230 231 def test_read_wrong_marks(self): 232 b = 'FORM' + struct.pack('>L', 4) + 'AIFF' 233 b += 'COMM' + struct.pack('>LhlhhLL', 18, 0, 0, 0, 0, 0, 0) 234 b += 'SSND' + struct.pack('>L', 8) + '\x00' * 8 235 b += 'MARK' + struct.pack('>LhB', 3, 1, 1) 236 with captured_stdout() as s: 237 f = aifc.open(io.BytesIO(b)) 238 self.assertEqual(s.getvalue(), 'Warning: MARK chunk contains ' 239 'only 0 markers instead of 1\n') 240 self.assertEqual(f.getmarkers(), None) 241 242 def test_read_comm_kludge_compname_even(self): 243 b = 'FORM' + struct.pack('>L', 4) + 'AIFC' 244 b += 'COMM' + struct.pack('>LhlhhLL', 18, 0, 0, 0, 0, 0, 0) 245 b += 'NONE' + struct.pack('B', 4) + 'even' + '\x00' 246 b += 'SSND' + struct.pack('>L', 8) + '\x00' * 8 247 with captured_stdout() as s: 248 f = aifc.open(io.BytesIO(b)) 249 self.assertEqual(s.getvalue(), 'Warning: bad COMM chunk size\n') 250 self.assertEqual(f.getcompname(), 'even') 251 252 def test_read_comm_kludge_compname_odd(self): 253 b = 'FORM' + struct.pack('>L', 4) + 'AIFC' 254 b += 'COMM' + struct.pack('>LhlhhLL', 18, 0, 0, 0, 0, 0, 0) 255 b += 'NONE' + struct.pack('B', 3) + 'odd' 256 b += 'SSND' + struct.pack('>L', 8) + '\x00' * 8 257 with captured_stdout() as s: 258 f = aifc.open(io.BytesIO(b)) 259 self.assertEqual(s.getvalue(), 'Warning: bad COMM chunk size\n') 260 self.assertEqual(f.getcompname(), 'odd') 261 262 def test_write_params_raises(self): 263 fout = aifc.open(io.BytesIO(), 'wb') 264 wrong_params = (0, 0, 0, 0, 'WRNG', '') 265 self.assertRaises(aifc.Error, fout.setparams, wrong_params) 266 self.assertRaises(aifc.Error, fout.getparams) 267 self.assertRaises(aifc.Error, fout.setnchannels, 0) 268 self.assertRaises(aifc.Error, fout.getnchannels) 269 self.assertRaises(aifc.Error, fout.setsampwidth, 0) 270 self.assertRaises(aifc.Error, fout.getsampwidth) 271 self.assertRaises(aifc.Error, fout.setframerate, 0) 272 self.assertRaises(aifc.Error, fout.getframerate) 273 self.assertRaises(aifc.Error, fout.setcomptype, 'WRNG', '') 274 fout.aiff() 275 fout.setnchannels(1) 276 fout.setsampwidth(1) 277 fout.setframerate(1) 278 fout.setnframes(1) 279 fout.writeframes('\x00') 280 self.assertRaises(aifc.Error, fout.setparams, (1, 1, 1, 1, 1, 1)) 281 self.assertRaises(aifc.Error, fout.setnchannels, 1) 282 self.assertRaises(aifc.Error, fout.setsampwidth, 1) 283 self.assertRaises(aifc.Error, fout.setframerate, 1) 284 self.assertRaises(aifc.Error, fout.setnframes, 1) 285 self.assertRaises(aifc.Error, fout.setcomptype, 'NONE', '') 286 self.assertRaises(aifc.Error, fout.aiff) 287 self.assertRaises(aifc.Error, fout.aifc) 288 289 def test_write_params_singles(self): 290 fout = aifc.open(io.BytesIO(), 'wb') 291 fout.aifc() 292 fout.setnchannels(1) 293 fout.setsampwidth(2) 294 fout.setframerate(3) 295 fout.setnframes(4) 296 fout.setcomptype('NONE', 'name') 297 self.assertEqual(fout.getnchannels(), 1) 298 self.assertEqual(fout.getsampwidth(), 2) 299 self.assertEqual(fout.getframerate(), 3) 300 self.assertEqual(fout.getnframes(), 0) 301 self.assertEqual(fout.tell(), 0) 302 self.assertEqual(fout.getcomptype(), 'NONE') 303 self.assertEqual(fout.getcompname(), 'name') 304 fout.writeframes('\x00' * 4 * fout.getsampwidth() * fout.getnchannels()) 305 self.assertEqual(fout.getnframes(), 4) 306 self.assertEqual(fout.tell(), 4) 307 308 def test_write_params_bunch(self): 309 fout = aifc.open(io.BytesIO(), 'wb') 310 fout.aifc() 311 p = (1, 2, 3, 4, 'NONE', 'name') 312 fout.setparams(p) 313 self.assertEqual(fout.getparams(), p) 314 fout.initfp(None) 315 316 def test_write_header_raises(self): 317 fout = aifc.open(io.BytesIO(), 'wb') 318 self.assertRaises(aifc.Error, fout.close) 319 fout = aifc.open(io.BytesIO(), 'wb') 320 fout.setnchannels(1) 321 self.assertRaises(aifc.Error, fout.close) 322 fout = aifc.open(io.BytesIO(), 'wb') 323 fout.setnchannels(1) 324 fout.setsampwidth(1) 325 self.assertRaises(aifc.Error, fout.close) 326 327 def test_write_header_comptype_raises(self): 328 for comptype in ('ULAW', 'ulaw', 'ALAW', 'alaw', 'G722'): 329 fout = aifc.open(io.BytesIO(), 'wb') 330 fout.setsampwidth(1) 331 fout.setcomptype(comptype, '') 332 self.assertRaises(aifc.Error, fout.close) 333 fout.initfp(None) 334 335 def test_write_markers_raises(self): 336 fout = aifc.open(io.BytesIO(), 'wb') 337 self.assertRaises(aifc.Error, fout.setmark, 0, 0, '') 338 self.assertRaises(aifc.Error, fout.setmark, 1, -1, '') 339 self.assertRaises(aifc.Error, fout.setmark, 1, 0, None) 340 self.assertRaises(aifc.Error, fout.getmark, 1) 341 fout.initfp(None) 342 343 def test_write_aiff_by_extension(self): 344 sampwidth = 2 345 filename = TESTFN + '.aiff' 346 self.addCleanup(unlink, filename) 347 348 fout = self.fout = aifc.open(filename, 'wb') 349 fout.setparams((1, sampwidth, 1, 1, 'ULAW', '')) 350 frames = '\x00' * fout.getnchannels() * sampwidth 351 fout.writeframes(frames) 352 fout.close() 353 354 f = self.f = aifc.open(filename, 'rb') 355 self.assertEqual(f.getcomptype(), 'NONE') 356 f.close() 357 358 359 def test_main(): 360 run_unittest(AifcPCM8Test, AifcPCM16Test, AifcPCM16Test, AifcPCM24Test, 361 AifcPCM32Test, AifcULAWTest, 362 AifcMiscTest, AIFCLowLevelTest) 363 364 if __name__ == "__main__": 365 test_main() 366
