test_audioop.py 28 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568
  1. import sys
  2. from test.support import warnings_helper
  3. import unittest
  4. audioop = warnings_helper.import_deprecated("audioop")
  5. def pack(width, data):
  6. return b''.join(v.to_bytes(width, sys.byteorder, signed=True) for v in data)
  7. def unpack(width, data):
  8. return [int.from_bytes(data[i: i + width], sys.byteorder, signed=True)
  9. for i in range(0, len(data), width)]
  10. packs = {w: (lambda *data, width=w: pack(width, data)) for w in (1, 2, 3, 4)}
  11. maxvalues = {w: (1 << (8 * w - 1)) - 1 for w in (1, 2, 3, 4)}
  12. minvalues = {w: -1 << (8 * w - 1) for w in (1, 2, 3, 4)}
  13. datas = {
  14. 1: b'\x00\x12\x45\xbb\x7f\x80\xff',
  15. 2: packs[2](0, 0x1234, 0x4567, -0x4567, 0x7fff, -0x8000, -1),
  16. 3: packs[3](0, 0x123456, 0x456789, -0x456789, 0x7fffff, -0x800000, -1),
  17. 4: packs[4](0, 0x12345678, 0x456789ab, -0x456789ab,
  18. 0x7fffffff, -0x80000000, -1),
  19. }
  20. INVALID_DATA = [
  21. (b'abc', 0),
  22. (b'abc', 2),
  23. (b'ab', 3),
  24. (b'abc', 4),
  25. ]
  26. class TestAudioop(unittest.TestCase):
  27. def test_max(self):
  28. for w in 1, 2, 3, 4:
  29. self.assertEqual(audioop.max(b'', w), 0)
  30. self.assertEqual(audioop.max(bytearray(), w), 0)
  31. self.assertEqual(audioop.max(memoryview(b''), w), 0)
  32. p = packs[w]
  33. self.assertEqual(audioop.max(p(5), w), 5)
  34. self.assertEqual(audioop.max(p(5, -8, -1), w), 8)
  35. self.assertEqual(audioop.max(p(maxvalues[w]), w), maxvalues[w])
  36. self.assertEqual(audioop.max(p(minvalues[w]), w), -minvalues[w])
  37. self.assertEqual(audioop.max(datas[w], w), -minvalues[w])
  38. def test_minmax(self):
  39. for w in 1, 2, 3, 4:
  40. self.assertEqual(audioop.minmax(b'', w),
  41. (0x7fffffff, -0x80000000))
  42. self.assertEqual(audioop.minmax(bytearray(), w),
  43. (0x7fffffff, -0x80000000))
  44. self.assertEqual(audioop.minmax(memoryview(b''), w),
  45. (0x7fffffff, -0x80000000))
  46. p = packs[w]
  47. self.assertEqual(audioop.minmax(p(5), w), (5, 5))
  48. self.assertEqual(audioop.minmax(p(5, -8, -1), w), (-8, 5))
  49. self.assertEqual(audioop.minmax(p(maxvalues[w]), w),
  50. (maxvalues[w], maxvalues[w]))
  51. self.assertEqual(audioop.minmax(p(minvalues[w]), w),
  52. (minvalues[w], minvalues[w]))
  53. self.assertEqual(audioop.minmax(datas[w], w),
  54. (minvalues[w], maxvalues[w]))
  55. def test_maxpp(self):
  56. for w in 1, 2, 3, 4:
  57. self.assertEqual(audioop.maxpp(b'', w), 0)
  58. self.assertEqual(audioop.maxpp(bytearray(), w), 0)
  59. self.assertEqual(audioop.maxpp(memoryview(b''), w), 0)
  60. self.assertEqual(audioop.maxpp(packs[w](*range(100)), w), 0)
  61. self.assertEqual(audioop.maxpp(packs[w](9, 10, 5, 5, 0, 1), w), 10)
  62. self.assertEqual(audioop.maxpp(datas[w], w),
  63. maxvalues[w] - minvalues[w])
  64. def test_avg(self):
  65. for w in 1, 2, 3, 4:
  66. self.assertEqual(audioop.avg(b'', w), 0)
  67. self.assertEqual(audioop.avg(bytearray(), w), 0)
  68. self.assertEqual(audioop.avg(memoryview(b''), w), 0)
  69. p = packs[w]
  70. self.assertEqual(audioop.avg(p(5), w), 5)
  71. self .assertEqual(audioop.avg(p(5, 8), w), 6)
  72. self.assertEqual(audioop.avg(p(5, -8), w), -2)
  73. self.assertEqual(audioop.avg(p(maxvalues[w], maxvalues[w]), w),
  74. maxvalues[w])
  75. self.assertEqual(audioop.avg(p(minvalues[w], minvalues[w]), w),
  76. minvalues[w])
  77. self.assertEqual(audioop.avg(packs[4](0x50000000, 0x70000000), 4),
  78. 0x60000000)
  79. self.assertEqual(audioop.avg(packs[4](-0x50000000, -0x70000000), 4),
  80. -0x60000000)
  81. def test_avgpp(self):
  82. for w in 1, 2, 3, 4:
  83. self.assertEqual(audioop.avgpp(b'', w), 0)
  84. self.assertEqual(audioop.avgpp(bytearray(), w), 0)
  85. self.assertEqual(audioop.avgpp(memoryview(b''), w), 0)
  86. self.assertEqual(audioop.avgpp(packs[w](*range(100)), w), 0)
  87. self.assertEqual(audioop.avgpp(packs[w](9, 10, 5, 5, 0, 1), w), 10)
  88. self.assertEqual(audioop.avgpp(datas[1], 1), 196)
  89. self.assertEqual(audioop.avgpp(datas[2], 2), 50534)
  90. self.assertEqual(audioop.avgpp(datas[3], 3), 12937096)
  91. self.assertEqual(audioop.avgpp(datas[4], 4), 3311897002)
  92. def test_rms(self):
  93. for w in 1, 2, 3, 4:
  94. self.assertEqual(audioop.rms(b'', w), 0)
  95. self.assertEqual(audioop.rms(bytearray(), w), 0)
  96. self.assertEqual(audioop.rms(memoryview(b''), w), 0)
  97. p = packs[w]
  98. self.assertEqual(audioop.rms(p(*range(100)), w), 57)
  99. self.assertAlmostEqual(audioop.rms(p(maxvalues[w]) * 5, w),
  100. maxvalues[w], delta=1)
  101. self.assertAlmostEqual(audioop.rms(p(minvalues[w]) * 5, w),
  102. -minvalues[w], delta=1)
  103. self.assertEqual(audioop.rms(datas[1], 1), 77)
  104. self.assertEqual(audioop.rms(datas[2], 2), 20001)
  105. self.assertEqual(audioop.rms(datas[3], 3), 5120523)
  106. self.assertEqual(audioop.rms(datas[4], 4), 1310854152)
  107. def test_cross(self):
  108. for w in 1, 2, 3, 4:
  109. self.assertEqual(audioop.cross(b'', w), -1)
  110. self.assertEqual(audioop.cross(bytearray(), w), -1)
  111. self.assertEqual(audioop.cross(memoryview(b''), w), -1)
  112. p = packs[w]
  113. self.assertEqual(audioop.cross(p(0, 1, 2), w), 0)
  114. self.assertEqual(audioop.cross(p(1, 2, -3, -4), w), 1)
  115. self.assertEqual(audioop.cross(p(-1, -2, 3, 4), w), 1)
  116. self.assertEqual(audioop.cross(p(0, minvalues[w]), w), 1)
  117. self.assertEqual(audioop.cross(p(minvalues[w], maxvalues[w]), w), 1)
  118. def test_add(self):
  119. for w in 1, 2, 3, 4:
  120. self.assertEqual(audioop.add(b'', b'', w), b'')
  121. self.assertEqual(audioop.add(bytearray(), bytearray(), w), b'')
  122. self.assertEqual(audioop.add(memoryview(b''), memoryview(b''), w), b'')
  123. self.assertEqual(audioop.add(datas[w], b'\0' * len(datas[w]), w),
  124. datas[w])
  125. self.assertEqual(audioop.add(datas[1], datas[1], 1),
  126. b'\x00\x24\x7f\x80\x7f\x80\xfe')
  127. self.assertEqual(audioop.add(datas[2], datas[2], 2),
  128. packs[2](0, 0x2468, 0x7fff, -0x8000, 0x7fff, -0x8000, -2))
  129. self.assertEqual(audioop.add(datas[3], datas[3], 3),
  130. packs[3](0, 0x2468ac, 0x7fffff, -0x800000,
  131. 0x7fffff, -0x800000, -2))
  132. self.assertEqual(audioop.add(datas[4], datas[4], 4),
  133. packs[4](0, 0x2468acf0, 0x7fffffff, -0x80000000,
  134. 0x7fffffff, -0x80000000, -2))
  135. def test_bias(self):
  136. for w in 1, 2, 3, 4:
  137. for bias in 0, 1, -1, 127, -128, 0x7fffffff, -0x80000000:
  138. self.assertEqual(audioop.bias(b'', w, bias), b'')
  139. self.assertEqual(audioop.bias(bytearray(), w, bias), b'')
  140. self.assertEqual(audioop.bias(memoryview(b''), w, bias), b'')
  141. self.assertEqual(audioop.bias(datas[1], 1, 1),
  142. b'\x01\x13\x46\xbc\x80\x81\x00')
  143. self.assertEqual(audioop.bias(datas[1], 1, -1),
  144. b'\xff\x11\x44\xba\x7e\x7f\xfe')
  145. self.assertEqual(audioop.bias(datas[1], 1, 0x7fffffff),
  146. b'\xff\x11\x44\xba\x7e\x7f\xfe')
  147. self.assertEqual(audioop.bias(datas[1], 1, -0x80000000),
  148. datas[1])
  149. self.assertEqual(audioop.bias(datas[2], 2, 1),
  150. packs[2](1, 0x1235, 0x4568, -0x4566, -0x8000, -0x7fff, 0))
  151. self.assertEqual(audioop.bias(datas[2], 2, -1),
  152. packs[2](-1, 0x1233, 0x4566, -0x4568, 0x7ffe, 0x7fff, -2))
  153. self.assertEqual(audioop.bias(datas[2], 2, 0x7fffffff),
  154. packs[2](-1, 0x1233, 0x4566, -0x4568, 0x7ffe, 0x7fff, -2))
  155. self.assertEqual(audioop.bias(datas[2], 2, -0x80000000),
  156. datas[2])
  157. self.assertEqual(audioop.bias(datas[3], 3, 1),
  158. packs[3](1, 0x123457, 0x45678a, -0x456788,
  159. -0x800000, -0x7fffff, 0))
  160. self.assertEqual(audioop.bias(datas[3], 3, -1),
  161. packs[3](-1, 0x123455, 0x456788, -0x45678a,
  162. 0x7ffffe, 0x7fffff, -2))
  163. self.assertEqual(audioop.bias(datas[3], 3, 0x7fffffff),
  164. packs[3](-1, 0x123455, 0x456788, -0x45678a,
  165. 0x7ffffe, 0x7fffff, -2))
  166. self.assertEqual(audioop.bias(datas[3], 3, -0x80000000),
  167. datas[3])
  168. self.assertEqual(audioop.bias(datas[4], 4, 1),
  169. packs[4](1, 0x12345679, 0x456789ac, -0x456789aa,
  170. -0x80000000, -0x7fffffff, 0))
  171. self.assertEqual(audioop.bias(datas[4], 4, -1),
  172. packs[4](-1, 0x12345677, 0x456789aa, -0x456789ac,
  173. 0x7ffffffe, 0x7fffffff, -2))
  174. self.assertEqual(audioop.bias(datas[4], 4, 0x7fffffff),
  175. packs[4](0x7fffffff, -0x6dcba989, -0x3a987656, 0x3a987654,
  176. -2, -1, 0x7ffffffe))
  177. self.assertEqual(audioop.bias(datas[4], 4, -0x80000000),
  178. packs[4](-0x80000000, -0x6dcba988, -0x3a987655, 0x3a987655,
  179. -1, 0, 0x7fffffff))
  180. def test_lin2lin(self):
  181. for w in 1, 2, 3, 4:
  182. self.assertEqual(audioop.lin2lin(datas[w], w, w), datas[w])
  183. self.assertEqual(audioop.lin2lin(bytearray(datas[w]), w, w),
  184. datas[w])
  185. self.assertEqual(audioop.lin2lin(memoryview(datas[w]), w, w),
  186. datas[w])
  187. self.assertEqual(audioop.lin2lin(datas[1], 1, 2),
  188. packs[2](0, 0x1200, 0x4500, -0x4500, 0x7f00, -0x8000, -0x100))
  189. self.assertEqual(audioop.lin2lin(datas[1], 1, 3),
  190. packs[3](0, 0x120000, 0x450000, -0x450000,
  191. 0x7f0000, -0x800000, -0x10000))
  192. self.assertEqual(audioop.lin2lin(datas[1], 1, 4),
  193. packs[4](0, 0x12000000, 0x45000000, -0x45000000,
  194. 0x7f000000, -0x80000000, -0x1000000))
  195. self.assertEqual(audioop.lin2lin(datas[2], 2, 1),
  196. b'\x00\x12\x45\xba\x7f\x80\xff')
  197. self.assertEqual(audioop.lin2lin(datas[2], 2, 3),
  198. packs[3](0, 0x123400, 0x456700, -0x456700,
  199. 0x7fff00, -0x800000, -0x100))
  200. self.assertEqual(audioop.lin2lin(datas[2], 2, 4),
  201. packs[4](0, 0x12340000, 0x45670000, -0x45670000,
  202. 0x7fff0000, -0x80000000, -0x10000))
  203. self.assertEqual(audioop.lin2lin(datas[3], 3, 1),
  204. b'\x00\x12\x45\xba\x7f\x80\xff')
  205. self.assertEqual(audioop.lin2lin(datas[3], 3, 2),
  206. packs[2](0, 0x1234, 0x4567, -0x4568, 0x7fff, -0x8000, -1))
  207. self.assertEqual(audioop.lin2lin(datas[3], 3, 4),
  208. packs[4](0, 0x12345600, 0x45678900, -0x45678900,
  209. 0x7fffff00, -0x80000000, -0x100))
  210. self.assertEqual(audioop.lin2lin(datas[4], 4, 1),
  211. b'\x00\x12\x45\xba\x7f\x80\xff')
  212. self.assertEqual(audioop.lin2lin(datas[4], 4, 2),
  213. packs[2](0, 0x1234, 0x4567, -0x4568, 0x7fff, -0x8000, -1))
  214. self.assertEqual(audioop.lin2lin(datas[4], 4, 3),
  215. packs[3](0, 0x123456, 0x456789, -0x45678a,
  216. 0x7fffff, -0x800000, -1))
  217. def test_adpcm2lin(self):
  218. self.assertEqual(audioop.adpcm2lin(b'\x07\x7f\x7f', 1, None),
  219. (b'\x00\x00\x00\xff\x00\xff', (-179, 40)))
  220. self.assertEqual(audioop.adpcm2lin(bytearray(b'\x07\x7f\x7f'), 1, None),
  221. (b'\x00\x00\x00\xff\x00\xff', (-179, 40)))
  222. self.assertEqual(audioop.adpcm2lin(memoryview(b'\x07\x7f\x7f'), 1, None),
  223. (b'\x00\x00\x00\xff\x00\xff', (-179, 40)))
  224. self.assertEqual(audioop.adpcm2lin(b'\x07\x7f\x7f', 2, None),
  225. (packs[2](0, 0xb, 0x29, -0x16, 0x72, -0xb3), (-179, 40)))
  226. self.assertEqual(audioop.adpcm2lin(b'\x07\x7f\x7f', 3, None),
  227. (packs[3](0, 0xb00, 0x2900, -0x1600, 0x7200,
  228. -0xb300), (-179, 40)))
  229. self.assertEqual(audioop.adpcm2lin(b'\x07\x7f\x7f', 4, None),
  230. (packs[4](0, 0xb0000, 0x290000, -0x160000, 0x720000,
  231. -0xb30000), (-179, 40)))
  232. # Very cursory test
  233. for w in 1, 2, 3, 4:
  234. self.assertEqual(audioop.adpcm2lin(b'\0' * 5, w, None),
  235. (b'\0' * w * 10, (0, 0)))
  236. def test_lin2adpcm(self):
  237. self.assertEqual(audioop.lin2adpcm(datas[1], 1, None),
  238. (b'\x07\x7f\x7f', (-221, 39)))
  239. self.assertEqual(audioop.lin2adpcm(bytearray(datas[1]), 1, None),
  240. (b'\x07\x7f\x7f', (-221, 39)))
  241. self.assertEqual(audioop.lin2adpcm(memoryview(datas[1]), 1, None),
  242. (b'\x07\x7f\x7f', (-221, 39)))
  243. for w in 2, 3, 4:
  244. self.assertEqual(audioop.lin2adpcm(datas[w], w, None),
  245. (b'\x07\x7f\x7f', (31, 39)))
  246. # Very cursory test
  247. for w in 1, 2, 3, 4:
  248. self.assertEqual(audioop.lin2adpcm(b'\0' * w * 10, w, None),
  249. (b'\0' * 5, (0, 0)))
  250. def test_invalid_adpcm_state(self):
  251. # state must be a tuple or None, not an integer
  252. self.assertRaises(TypeError, audioop.adpcm2lin, b'\0', 1, 555)
  253. self.assertRaises(TypeError, audioop.lin2adpcm, b'\0', 1, 555)
  254. # Issues #24456, #24457: index out of range
  255. self.assertRaises(ValueError, audioop.adpcm2lin, b'\0', 1, (0, -1))
  256. self.assertRaises(ValueError, audioop.adpcm2lin, b'\0', 1, (0, 89))
  257. self.assertRaises(ValueError, audioop.lin2adpcm, b'\0', 1, (0, -1))
  258. self.assertRaises(ValueError, audioop.lin2adpcm, b'\0', 1, (0, 89))
  259. # value out of range
  260. self.assertRaises(ValueError, audioop.adpcm2lin, b'\0', 1, (-0x8001, 0))
  261. self.assertRaises(ValueError, audioop.adpcm2lin, b'\0', 1, (0x8000, 0))
  262. self.assertRaises(ValueError, audioop.lin2adpcm, b'\0', 1, (-0x8001, 0))
  263. self.assertRaises(ValueError, audioop.lin2adpcm, b'\0', 1, (0x8000, 0))
  264. def test_lin2alaw(self):
  265. self.assertEqual(audioop.lin2alaw(datas[1], 1),
  266. b'\xd5\x87\xa4\x24\xaa\x2a\x5a')
  267. self.assertEqual(audioop.lin2alaw(bytearray(datas[1]), 1),
  268. b'\xd5\x87\xa4\x24\xaa\x2a\x5a')
  269. self.assertEqual(audioop.lin2alaw(memoryview(datas[1]), 1),
  270. b'\xd5\x87\xa4\x24\xaa\x2a\x5a')
  271. for w in 2, 3, 4:
  272. self.assertEqual(audioop.lin2alaw(datas[w], w),
  273. b'\xd5\x87\xa4\x24\xaa\x2a\x55')
  274. def test_alaw2lin(self):
  275. encoded = b'\x00\x03\x24\x2a\x51\x54\x55\x58\x6b\x71\x7f'\
  276. b'\x80\x83\xa4\xaa\xd1\xd4\xd5\xd8\xeb\xf1\xff'
  277. src = [-688, -720, -2240, -4032, -9, -3, -1, -27, -244, -82, -106,
  278. 688, 720, 2240, 4032, 9, 3, 1, 27, 244, 82, 106]
  279. for w in 1, 2, 3, 4:
  280. decoded = packs[w](*(x << (w * 8) >> 13 for x in src))
  281. self.assertEqual(audioop.alaw2lin(encoded, w), decoded)
  282. self.assertEqual(audioop.alaw2lin(bytearray(encoded), w), decoded)
  283. self.assertEqual(audioop.alaw2lin(memoryview(encoded), w), decoded)
  284. encoded = bytes(range(256))
  285. for w in 2, 3, 4:
  286. decoded = audioop.alaw2lin(encoded, w)
  287. self.assertEqual(audioop.lin2alaw(decoded, w), encoded)
  288. def test_lin2ulaw(self):
  289. self.assertEqual(audioop.lin2ulaw(datas[1], 1),
  290. b'\xff\xad\x8e\x0e\x80\x00\x67')
  291. self.assertEqual(audioop.lin2ulaw(bytearray(datas[1]), 1),
  292. b'\xff\xad\x8e\x0e\x80\x00\x67')
  293. self.assertEqual(audioop.lin2ulaw(memoryview(datas[1]), 1),
  294. b'\xff\xad\x8e\x0e\x80\x00\x67')
  295. for w in 2, 3, 4:
  296. self.assertEqual(audioop.lin2ulaw(datas[w], w),
  297. b'\xff\xad\x8e\x0e\x80\x00\x7e')
  298. def test_ulaw2lin(self):
  299. encoded = b'\x00\x0e\x28\x3f\x57\x6a\x76\x7c\x7e\x7f'\
  300. b'\x80\x8e\xa8\xbf\xd7\xea\xf6\xfc\xfe\xff'
  301. src = [-8031, -4447, -1471, -495, -163, -53, -18, -6, -2, 0,
  302. 8031, 4447, 1471, 495, 163, 53, 18, 6, 2, 0]
  303. for w in 1, 2, 3, 4:
  304. decoded = packs[w](*(x << (w * 8) >> 14 for x in src))
  305. self.assertEqual(audioop.ulaw2lin(encoded, w), decoded)
  306. self.assertEqual(audioop.ulaw2lin(bytearray(encoded), w), decoded)
  307. self.assertEqual(audioop.ulaw2lin(memoryview(encoded), w), decoded)
  308. # Current u-law implementation has two codes fo 0: 0x7f and 0xff.
  309. encoded = bytes(range(127)) + bytes(range(128, 256))
  310. for w in 2, 3, 4:
  311. decoded = audioop.ulaw2lin(encoded, w)
  312. self.assertEqual(audioop.lin2ulaw(decoded, w), encoded)
  313. def test_mul(self):
  314. for w in 1, 2, 3, 4:
  315. self.assertEqual(audioop.mul(b'', w, 2), b'')
  316. self.assertEqual(audioop.mul(bytearray(), w, 2), b'')
  317. self.assertEqual(audioop.mul(memoryview(b''), w, 2), b'')
  318. self.assertEqual(audioop.mul(datas[w], w, 0),
  319. b'\0' * len(datas[w]))
  320. self.assertEqual(audioop.mul(datas[w], w, 1),
  321. datas[w])
  322. self.assertEqual(audioop.mul(datas[1], 1, 2),
  323. b'\x00\x24\x7f\x80\x7f\x80\xfe')
  324. self.assertEqual(audioop.mul(datas[2], 2, 2),
  325. packs[2](0, 0x2468, 0x7fff, -0x8000, 0x7fff, -0x8000, -2))
  326. self.assertEqual(audioop.mul(datas[3], 3, 2),
  327. packs[3](0, 0x2468ac, 0x7fffff, -0x800000,
  328. 0x7fffff, -0x800000, -2))
  329. self.assertEqual(audioop.mul(datas[4], 4, 2),
  330. packs[4](0, 0x2468acf0, 0x7fffffff, -0x80000000,
  331. 0x7fffffff, -0x80000000, -2))
  332. def test_ratecv(self):
  333. for w in 1, 2, 3, 4:
  334. self.assertEqual(audioop.ratecv(b'', w, 1, 8000, 8000, None),
  335. (b'', (-1, ((0, 0),))))
  336. self.assertEqual(audioop.ratecv(bytearray(), w, 1, 8000, 8000, None),
  337. (b'', (-1, ((0, 0),))))
  338. self.assertEqual(audioop.ratecv(memoryview(b''), w, 1, 8000, 8000, None),
  339. (b'', (-1, ((0, 0),))))
  340. self.assertEqual(audioop.ratecv(b'', w, 5, 8000, 8000, None),
  341. (b'', (-1, ((0, 0),) * 5)))
  342. self.assertEqual(audioop.ratecv(b'', w, 1, 8000, 16000, None),
  343. (b'', (-2, ((0, 0),))))
  344. self.assertEqual(audioop.ratecv(datas[w], w, 1, 8000, 8000, None)[0],
  345. datas[w])
  346. self.assertEqual(audioop.ratecv(datas[w], w, 1, 8000, 8000, None, 1, 0)[0],
  347. datas[w])
  348. state = None
  349. d1, state = audioop.ratecv(b'\x00\x01\x02', 1, 1, 8000, 16000, state)
  350. d2, state = audioop.ratecv(b'\x00\x01\x02', 1, 1, 8000, 16000, state)
  351. self.assertEqual(d1 + d2, b'\000\000\001\001\002\001\000\000\001\001\002')
  352. for w in 1, 2, 3, 4:
  353. d0, state0 = audioop.ratecv(datas[w], w, 1, 8000, 16000, None)
  354. d, state = b'', None
  355. for i in range(0, len(datas[w]), w):
  356. d1, state = audioop.ratecv(datas[w][i:i + w], w, 1,
  357. 8000, 16000, state)
  358. d += d1
  359. self.assertEqual(d, d0)
  360. self.assertEqual(state, state0)
  361. expected = {
  362. 1: packs[1](0, 0x0d, 0x37, -0x26, 0x55, -0x4b, -0x14),
  363. 2: packs[2](0, 0x0da7, 0x3777, -0x2630, 0x5673, -0x4a64, -0x129a),
  364. 3: packs[3](0, 0x0da740, 0x377776, -0x262fca,
  365. 0x56740c, -0x4a62fd, -0x1298c0),
  366. 4: packs[4](0, 0x0da740da, 0x37777776, -0x262fc962,
  367. 0x56740da6, -0x4a62fc96, -0x1298bf26),
  368. }
  369. for w in 1, 2, 3, 4:
  370. self.assertEqual(audioop.ratecv(datas[w], w, 1, 8000, 8000, None, 3, 1)[0],
  371. expected[w])
  372. self.assertEqual(audioop.ratecv(datas[w], w, 1, 8000, 8000, None, 30, 10)[0],
  373. expected[w])
  374. self.assertRaises(TypeError, audioop.ratecv, b'', 1, 1, 8000, 8000, 42)
  375. self.assertRaises(TypeError, audioop.ratecv,
  376. b'', 1, 1, 8000, 8000, (1, (42,)))
  377. def test_reverse(self):
  378. for w in 1, 2, 3, 4:
  379. self.assertEqual(audioop.reverse(b'', w), b'')
  380. self.assertEqual(audioop.reverse(bytearray(), w), b'')
  381. self.assertEqual(audioop.reverse(memoryview(b''), w), b'')
  382. self.assertEqual(audioop.reverse(packs[w](0, 1, 2), w),
  383. packs[w](2, 1, 0))
  384. def test_tomono(self):
  385. for w in 1, 2, 3, 4:
  386. data1 = datas[w]
  387. data2 = bytearray(2 * len(data1))
  388. for k in range(w):
  389. data2[k::2*w] = data1[k::w]
  390. self.assertEqual(audioop.tomono(data2, w, 1, 0), data1)
  391. self.assertEqual(audioop.tomono(data2, w, 0, 1), b'\0' * len(data1))
  392. for k in range(w):
  393. data2[k+w::2*w] = data1[k::w]
  394. self.assertEqual(audioop.tomono(data2, w, 0.5, 0.5), data1)
  395. self.assertEqual(audioop.tomono(bytearray(data2), w, 0.5, 0.5),
  396. data1)
  397. self.assertEqual(audioop.tomono(memoryview(data2), w, 0.5, 0.5),
  398. data1)
  399. def test_tostereo(self):
  400. for w in 1, 2, 3, 4:
  401. data1 = datas[w]
  402. data2 = bytearray(2 * len(data1))
  403. for k in range(w):
  404. data2[k::2*w] = data1[k::w]
  405. self.assertEqual(audioop.tostereo(data1, w, 1, 0), data2)
  406. self.assertEqual(audioop.tostereo(data1, w, 0, 0), b'\0' * len(data2))
  407. for k in range(w):
  408. data2[k+w::2*w] = data1[k::w]
  409. self.assertEqual(audioop.tostereo(data1, w, 1, 1), data2)
  410. self.assertEqual(audioop.tostereo(bytearray(data1), w, 1, 1), data2)
  411. self.assertEqual(audioop.tostereo(memoryview(data1), w, 1, 1),
  412. data2)
  413. def test_findfactor(self):
  414. self.assertEqual(audioop.findfactor(datas[2], datas[2]), 1.0)
  415. self.assertEqual(audioop.findfactor(bytearray(datas[2]),
  416. bytearray(datas[2])), 1.0)
  417. self.assertEqual(audioop.findfactor(memoryview(datas[2]),
  418. memoryview(datas[2])), 1.0)
  419. self.assertEqual(audioop.findfactor(b'\0' * len(datas[2]), datas[2]),
  420. 0.0)
  421. def test_findfit(self):
  422. self.assertEqual(audioop.findfit(datas[2], datas[2]), (0, 1.0))
  423. self.assertEqual(audioop.findfit(bytearray(datas[2]),
  424. bytearray(datas[2])), (0, 1.0))
  425. self.assertEqual(audioop.findfit(memoryview(datas[2]),
  426. memoryview(datas[2])), (0, 1.0))
  427. self.assertEqual(audioop.findfit(datas[2], packs[2](1, 2, 0)),
  428. (1, 8038.8))
  429. self.assertEqual(audioop.findfit(datas[2][:-2] * 5 + datas[2], datas[2]),
  430. (30, 1.0))
  431. def test_findmax(self):
  432. self.assertEqual(audioop.findmax(datas[2], 1), 5)
  433. self.assertEqual(audioop.findmax(bytearray(datas[2]), 1), 5)
  434. self.assertEqual(audioop.findmax(memoryview(datas[2]), 1), 5)
  435. def test_getsample(self):
  436. for w in 1, 2, 3, 4:
  437. data = packs[w](0, 1, -1, maxvalues[w], minvalues[w])
  438. self.assertEqual(audioop.getsample(data, w, 0), 0)
  439. self.assertEqual(audioop.getsample(bytearray(data), w, 0), 0)
  440. self.assertEqual(audioop.getsample(memoryview(data), w, 0), 0)
  441. self.assertEqual(audioop.getsample(data, w, 1), 1)
  442. self.assertEqual(audioop.getsample(data, w, 2), -1)
  443. self.assertEqual(audioop.getsample(data, w, 3), maxvalues[w])
  444. self.assertEqual(audioop.getsample(data, w, 4), minvalues[w])
  445. def test_byteswap(self):
  446. swapped_datas = {
  447. 1: datas[1],
  448. 2: packs[2](0, 0x3412, 0x6745, -0x6646, -0x81, 0x80, -1),
  449. 3: packs[3](0, 0x563412, -0x7698bb, 0x7798ba, -0x81, 0x80, -1),
  450. 4: packs[4](0, 0x78563412, -0x547698bb, 0x557698ba,
  451. -0x81, 0x80, -1),
  452. }
  453. for w in 1, 2, 3, 4:
  454. self.assertEqual(audioop.byteswap(b'', w), b'')
  455. self.assertEqual(audioop.byteswap(datas[w], w), swapped_datas[w])
  456. self.assertEqual(audioop.byteswap(swapped_datas[w], w), datas[w])
  457. self.assertEqual(audioop.byteswap(bytearray(datas[w]), w),
  458. swapped_datas[w])
  459. self.assertEqual(audioop.byteswap(memoryview(datas[w]), w),
  460. swapped_datas[w])
  461. def test_negativelen(self):
  462. # from issue 3306, previously it segfaulted
  463. self.assertRaises(audioop.error,
  464. audioop.findmax, bytes(range(256)), -2392392)
  465. def test_issue7673(self):
  466. state = None
  467. for data, size in INVALID_DATA:
  468. size2 = size
  469. self.assertRaises(audioop.error, audioop.getsample, data, size, 0)
  470. self.assertRaises(audioop.error, audioop.max, data, size)
  471. self.assertRaises(audioop.error, audioop.minmax, data, size)
  472. self.assertRaises(audioop.error, audioop.avg, data, size)
  473. self.assertRaises(audioop.error, audioop.rms, data, size)
  474. self.assertRaises(audioop.error, audioop.avgpp, data, size)
  475. self.assertRaises(audioop.error, audioop.maxpp, data, size)
  476. self.assertRaises(audioop.error, audioop.cross, data, size)
  477. self.assertRaises(audioop.error, audioop.mul, data, size, 1.0)
  478. self.assertRaises(audioop.error, audioop.tomono, data, size, 0.5, 0.5)
  479. self.assertRaises(audioop.error, audioop.tostereo, data, size, 0.5, 0.5)
  480. self.assertRaises(audioop.error, audioop.add, data, data, size)
  481. self.assertRaises(audioop.error, audioop.bias, data, size, 0)
  482. self.assertRaises(audioop.error, audioop.reverse, data, size)
  483. self.assertRaises(audioop.error, audioop.lin2lin, data, size, size2)
  484. self.assertRaises(audioop.error, audioop.ratecv, data, size, 1, 1, 1, state)
  485. self.assertRaises(audioop.error, audioop.lin2ulaw, data, size)
  486. self.assertRaises(audioop.error, audioop.lin2alaw, data, size)
  487. self.assertRaises(audioop.error, audioop.lin2adpcm, data, size, state)
  488. def test_string(self):
  489. data = 'abcd'
  490. size = 2
  491. self.assertRaises(TypeError, audioop.getsample, data, size, 0)
  492. self.assertRaises(TypeError, audioop.max, data, size)
  493. self.assertRaises(TypeError, audioop.minmax, data, size)
  494. self.assertRaises(TypeError, audioop.avg, data, size)
  495. self.assertRaises(TypeError, audioop.rms, data, size)
  496. self.assertRaises(TypeError, audioop.avgpp, data, size)
  497. self.assertRaises(TypeError, audioop.maxpp, data, size)
  498. self.assertRaises(TypeError, audioop.cross, data, size)
  499. self.assertRaises(TypeError, audioop.mul, data, size, 1.0)
  500. self.assertRaises(TypeError, audioop.tomono, data, size, 0.5, 0.5)
  501. self.assertRaises(TypeError, audioop.tostereo, data, size, 0.5, 0.5)
  502. self.assertRaises(TypeError, audioop.add, data, data, size)
  503. self.assertRaises(TypeError, audioop.bias, data, size, 0)
  504. self.assertRaises(TypeError, audioop.reverse, data, size)
  505. self.assertRaises(TypeError, audioop.lin2lin, data, size, size)
  506. self.assertRaises(TypeError, audioop.ratecv, data, size, 1, 1, 1, None)
  507. self.assertRaises(TypeError, audioop.lin2ulaw, data, size)
  508. self.assertRaises(TypeError, audioop.lin2alaw, data, size)
  509. self.assertRaises(TypeError, audioop.lin2adpcm, data, size, None)
  510. def test_wrongsize(self):
  511. data = b'abcdefgh'
  512. state = None
  513. for size in (-1, 0, 5, 1024):
  514. self.assertRaises(audioop.error, audioop.ulaw2lin, data, size)
  515. self.assertRaises(audioop.error, audioop.alaw2lin, data, size)
  516. self.assertRaises(audioop.error, audioop.adpcm2lin, data, size, state)
  517. if __name__ == '__main__':
  518. unittest.main()