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