Mercurial > hg > auditok
comparison tests/test_signal.py @ 359:63871bac77f6
Rename functions
author | Amine Sehili <amine.sehili@gmail.com> |
---|---|
date | Thu, 02 Apr 2020 22:29:43 +0200 |
parents | 379e36e4fe88 |
children | d653e3f58f3c |
comparison
equal
deleted
inserted
replaced
358:379e36e4fe88 | 359:63871bac77f6 |
---|---|
87 int8_4channel=("b", 4, [50, 54, 64]), | 87 int8_4channel=("b", 4, [50, 54, 64]), |
88 int16_1channel=("h", 1, [12592, 13106, 13620, 14134, 16697, 17218]), | 88 int16_1channel=("h", 1, [12592, 13106, 13620, 14134, 16697, 17218]), |
89 int16_2channel=("h", 2, [12849, 13877, 16958]), | 89 int16_2channel=("h", 2, [12849, 13877, 16958]), |
90 int32_3channel=("i", 3, [971214132]), | 90 int32_3channel=("i", 3, [971214132]), |
91 ) | 91 ) |
92 def test_average_channels(self, fmt, channels, expected): | 92 def test_compute_average_channel(self, fmt, channels, expected): |
93 result = signal_.average_channels(self.data, fmt, channels) | 93 result = signal_.compute_average_channel(self.data, fmt, channels) |
94 expected = array_(fmt, expected) | 94 expected = array_(fmt, expected) |
95 expected_numpy_fmt = self.numpy_fmt[fmt] | 95 expected_numpy_fmt = self.numpy_fmt[fmt] |
96 self.assertEqual(result, expected) | 96 self.assertEqual(result, expected) |
97 result_numpy = signal_numpy.average_channels( | 97 result_numpy = signal_numpy.compute_average_channel( |
98 self.data, self.numpy_fmt[fmt], channels | 98 self.data, self.numpy_fmt[fmt], channels |
99 ) | 99 ) |
100 self.assertTrue(all(result_numpy == expected)) | 100 self.assertTrue(all(result_numpy == expected)) |
101 self.assertEqual(result_numpy.dtype, expected_numpy_fmt) | 101 self.assertEqual(result_numpy.dtype, expected_numpy_fmt) |
102 | 102 |
103 @genty_dataset( | 103 @genty_dataset( |
104 int8_2channel=(1, [48, 50, 52, 54, 61, 66]), | 104 int8_2channel=(1, [48, 50, 52, 54, 61, 66]), |
105 int16_2channel=(2, [12849, 13877, 16957]), | 105 int16_2channel=(2, [12849, 13877, 16957]), |
106 ) | 106 ) |
107 def test_average_channels_stereo(self, sample_width, expected): | 107 def test_compute_average_channel_stereo(self, sample_width, expected): |
108 result = signal_.average_channels_stereo(self.data, sample_width) | 108 result = signal_.compute_average_channel_stereo( |
109 self.data, sample_width | |
110 ) | |
109 fmt = signal_.FORMAT[sample_width] | 111 fmt = signal_.FORMAT[sample_width] |
110 expected = array_(fmt, expected) | 112 expected = array_(fmt, expected) |
111 self.assertEqual(result, expected) | 113 self.assertEqual(result, expected) |
112 | 114 |
113 @genty_dataset( | 115 @genty_dataset( |