source: python/tests/test_slicing.py @ f0ce8aa3

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

[py] [slicing] add test for missing end of last region
  • Property mode set to 100755
File size: 7.3 KB
Line 
1#! /usr/bin/env python
2
3from unittest import main
4from numpy.testing import TestCase, assert_equal
5from aubio import slice_source_at_stamps
6from .utils import count_files_in_directory, get_default_test_sound
7from .utils import count_samples_in_directory, count_samples_in_file
8
9import tempfile
10import shutil
11
12n_slices = 4
13
14class aubio_slicing_test_case(TestCase):
15
16    def setUp(self):
17        self.source_file = get_default_test_sound(self)
18        self.output_dir = tempfile.mkdtemp(suffix = 'aubio_slicing_test_case')
19
20    def test_slice_start_only(self):
21        regions_start = [i*1000 for i in range(n_slices)]
22        slice_source_at_stamps(self.source_file, regions_start, output_dir = self.output_dir)
23
24    def test_slice_start_only_no_zero(self):
25        regions_start = [i*1000 for i in range(1, n_slices)]
26        slice_source_at_stamps(self.source_file, regions_start, output_dir = self.output_dir)
27
28    def test_slice_start_beyond_end(self):
29        regions_start = [i*1000 for i in range(1, n_slices)]
30        regions_start += [count_samples_in_file(self.source_file) + 1000]
31        slice_source_at_stamps(self.source_file, regions_start, output_dir = self.output_dir)
32
33    def test_slice_start_every_blocksize(self):
34        hopsize = 200
35        regions_start = [i*hopsize for i in range(1, n_slices)]
36        slice_source_at_stamps(self.source_file, regions_start, output_dir = self.output_dir,
37                hopsize = 200)
38
39    def test_slice_start_every_half_blocksize(self):
40        hopsize = 200
41        regions_start = [i*hopsize//2 for i in range(1, n_slices)]
42        slice_source_at_stamps(self.source_file, regions_start,
43                output_dir = self.output_dir, hopsize = 200)
44
45    def tearDown(self):
46        original_samples = count_samples_in_file(self.source_file)
47        written_samples = count_samples_in_directory(self.output_dir)
48        total_files = count_files_in_directory(self.output_dir)
49        assert_equal(n_slices, total_files,
50            "number of slices created different from expected")
51        assert_equal(written_samples, original_samples,
52            "number of samples written different from number of original samples")
53        shutil.rmtree(self.output_dir)
54
55class aubio_slicing_with_ends_test_case(TestCase):
56
57    def setUp(self):
58        self.source_file = get_default_test_sound(self)
59        self.output_dir = tempfile.mkdtemp(suffix = 'aubio_slicing_test_case')
60
61    def test_slice_start_and_ends_no_gap(self):
62        regions_start = [i*1000 for i in range(n_slices)]
63        regions_ends = [start - 1 for start in regions_start[1:]] + [1e120]
64        slice_source_at_stamps(self.source_file, regions_start, regions_ends,
65                output_dir = self.output_dir)
66        original_samples = count_samples_in_file(self.source_file)
67        written_samples = count_samples_in_directory(self.output_dir)
68        total_files = count_files_in_directory(self.output_dir)
69        assert_equal(n_slices, total_files,
70            "number of slices created different from expected")
71        assert_equal(written_samples, original_samples,
72            "number of samples written different from number of original samples")
73
74    def test_slice_start_and_ends_200_gap(self):
75        regions_start = [i*1000 for i in range(n_slices)]
76        regions_ends = [start + 199 for start in regions_start]
77        slice_source_at_stamps(self.source_file, regions_start, regions_ends,
78                output_dir = self.output_dir)
79        expected_samples = 200 * n_slices
80        written_samples = count_samples_in_directory(self.output_dir)
81        total_files = count_files_in_directory(self.output_dir)
82        assert_equal(n_slices, total_files,
83            "number of slices created different from expected")
84        assert_equal(written_samples, expected_samples,
85            "number of samples written different from number of original samples")
86
87    def test_slice_start_and_ends_overlaping(self):
88        regions_start = [i*1000 for i in range(n_slices)]
89        regions_ends = [start + 1199 for start in regions_start]
90        slice_source_at_stamps(self.source_file, regions_start, regions_ends,
91                output_dir = self.output_dir)
92        expected_samples = 1200 * n_slices
93        written_samples = count_samples_in_directory(self.output_dir)
94        total_files = count_files_in_directory(self.output_dir)
95        assert_equal(n_slices, total_files,
96            "number of slices created different from expected")
97        assert_equal(written_samples, expected_samples,
98            "number of samples written different from number of original samples")
99
100    def test_slice_start_and_ends_with_missing_end(self):
101        regions_start = [i*1000 for i in range(n_slices)]
102        regions_ends = [r-1 for r in regions_start[1:]]
103        slice_source_at_stamps(self.source_file, regions_start, regions_ends,
104                output_dir = self.output_dir)
105        written_samples = count_samples_in_directory(self.output_dir)
106        original_samples = count_samples_in_file(self.source_file)
107        total_files = count_files_in_directory(self.output_dir)
108        assert_equal(n_slices, total_files,
109            "number of slices created different from expected")
110        assert_equal(written_samples, original_samples,
111            "number of samples written different from number of original samples")
112
113    def tearDown(self):
114        shutil.rmtree(self.output_dir)
115
116
117class aubio_slicing_wrong_starts_test_case(TestCase):
118
119    def setUp(self):
120        self.source_file = get_default_test_sound(self)
121        self.output_dir = tempfile.mkdtemp(suffix = 'aubio_slicing_test_case')
122
123    def test_slice_start_empty(self):
124        regions_start = []
125        self.assertRaises(ValueError,
126                slice_source_at_stamps,
127                self.source_file, regions_start, output_dir = self.output_dir)
128
129    def test_slice_start_none(self):
130        regions_start = None
131        self.assertRaises(ValueError,
132                slice_source_at_stamps,
133                self.source_file, regions_start, output_dir = self.output_dir)
134
135    def tearDown(self):
136        shutil.rmtree(self.output_dir)
137
138class aubio_slicing_wrong_ends_test_case(TestCase):
139
140    def setUp(self):
141        self.source_file = get_default_test_sound(self)
142        self.output_dir = tempfile.mkdtemp(suffix = 'aubio_slicing_test_case')
143
144    def test_slice_wrong_ends(self):
145        regions_start = [i*1000 for i in range(1, n_slices)]
146        regions_end = []
147        self.assertRaises (ValueError,
148            slice_source_at_stamps, self.source_file, regions_start, regions_end,
149                output_dir = self.output_dir)
150
151    def test_slice_no_ends(self):
152        regions_start = [i*1000 for i in range(1, n_slices)]
153        regions_end = None
154        slice_source_at_stamps (self.source_file, regions_start, regions_end,
155                output_dir = self.output_dir)
156        total_files = count_files_in_directory(self.output_dir)
157        assert_equal(n_slices, total_files,
158            "number of slices created different from expected")
159        original_samples = count_samples_in_file(self.source_file)
160        written_samples = count_samples_in_directory(self.output_dir)
161        assert_equal(written_samples, original_samples,
162            "number of samples written different from number of original samples")
163
164    def tearDown(self):
165        shutil.rmtree(self.output_dir)
166
167if __name__ == '__main__':
168    main()
Note: See TracBrowser for help on using the repository browser.