source: python/tests/test_source_channels.py @ ca99325

feature/autosinkfeature/cnnfeature/cnn_orgfeature/constantqfeature/crepefeature/crepe_orgfeature/pitchshiftfeature/pydocstringsfeature/timestretchfix/ffmpeg5
Last change on this file since ca99325 was ca99325, checked in by Paul Brossier <piem@piem.org>, 6 years ago

[test] add a test using sink and source to check channel mixing

  • Property mode set to 100755
File size: 3.1 KB
Line 
1#! /usr/bin/env python
2
3"""A brute force test using `sink` to create and write samples to a stereo
4file, then `source` to check the correct content is read from the files."""
5
6import os.path
7import unittest
8import aubio
9import numpy as np
10from numpy.testing import assert_equal
11from utils import get_tmp_sink_path
12
13class aubio_source_test_case(unittest.TestCase):
14
15    def test_read_from_mono(self):
16        out = get_tmp_sink_path()
17        samplerate = 44100
18        hop_size = 256
19        blocks = 10
20        channels = 1
21        write_samples = np.ones([channels, hop_size], dtype=aubio.float_type)
22        write_samples *= .5
23        self.check_write_and_read(samplerate, channels, hop_size, blocks,
24                write_samples)
25
26    def test_read_from_stereo(self):
27        out = get_tmp_sink_path()
28        samplerate = 44100
29        hop_size = 256
30        blocks = 10
31        channels = 1
32        write_samples = np.ones([channels, hop_size], dtype=aubio.float_type)
33        write_samples *= .5
34        self.check_write_and_read(samplerate, channels, hop_size, blocks,
35                write_samples)
36
37    def test_read_from_half_stereo(self):
38        samplerate = 16000
39        channels = 2
40        hop_size = 512
41        blocks = 10
42        write_samples = np.ones([channels, hop_size], dtype=aubio.float_type)
43        write_samples *= .5
44        write_samples[1, :] = 0
45        self.check_write_and_read(samplerate, channels, hop_size, blocks,
46                write_samples)
47
48    def test_read_from_cancelling_channels(self):
49        samplerate = 16000
50        channels = 2
51        hop_size = 512
52        blocks = 10
53        write_samples = np.ones([channels, hop_size], dtype=aubio.float_type)
54        write_samples *= .5
55        write_samples[1] *= -1
56        self.check_write_and_read(samplerate, channels, hop_size, blocks,
57                write_samples)
58
59    def test_read_from_strange_three_channels(self):
60        samplerate = 8000
61        channels = 3
62        hop_size = 123
63        blocks = 10
64        write_samples = np.ones([channels, hop_size], dtype=aubio.float_type)
65        write_samples *= .5
66        write_samples[1, :] = 0
67        self.check_write_and_read(samplerate, channels, hop_size, blocks,
68                write_samples)
69
70    def check_write_and_read(self, samplerate, channels,
71            hop_size, blocks, write_samples):
72        expected_mono = np.sum(write_samples, axis=0)/write_samples.shape[0]
73        out = get_tmp_sink_path()
74        snk = aubio.sink(out, samplerate, channels=channels)
75        for i in range(blocks):
76            snk.do_multi(write_samples, hop_size)
77        # close the sink before reading from it
78        snk.close()
79
80        src = aubio.source(out, samplerate, hop_size)
81        for i in range(blocks):
82            read_samples, read = src.do_multi()
83            assert_equal (read_samples, write_samples)
84            assert_equal (read, hop_size)
85
86        src.seek(0)
87        for i in range(blocks):
88            read_samples, read = src()
89            assert_equal (read, hop_size)
90            assert_equal (read_samples, expected_mono)
91
92if __name__ == '__main__':
93    unittest.main()
Note: See TracBrowser for help on using the repository browser.