1 | #! /usr/bin/env python |
---|
2 | |
---|
3 | from numpy.testing import TestCase, run_module_suite |
---|
4 | from numpy.testing import assert_equal, assert_almost_equal |
---|
5 | |
---|
6 | from aubio import slice_source_at_stamps |
---|
7 | from utils import * |
---|
8 | |
---|
9 | import tempfile |
---|
10 | import shutil |
---|
11 | |
---|
12 | n_slices = 8 |
---|
13 | |
---|
14 | class 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 | regions_start += [count_samples_in_file(self.source_file) + 1000] |
---|
37 | slice_source_at_stamps(self.source_file, regions_start, output_dir = self.output_dir, |
---|
38 | hopsize = 200) |
---|
39 | |
---|
40 | def tearDown(self): |
---|
41 | original_samples = count_samples_in_file(self.source_file) |
---|
42 | written_samples = count_samples_in_directory(self.output_dir) |
---|
43 | total_files = count_files_in_directory(self.output_dir) |
---|
44 | assert_equal(n_slices, total_files, |
---|
45 | "number of slices created different from expected") |
---|
46 | assert_equal(written_samples, original_samples, |
---|
47 | "number of samples written different from number of original samples") |
---|
48 | shutil.rmtree(self.output_dir) |
---|
49 | |
---|
50 | class aubio_slicing_wrong_starts_test_case(TestCase): |
---|
51 | |
---|
52 | def setUp(self): |
---|
53 | self.source_file = get_default_test_sound(self) |
---|
54 | self.output_dir = tempfile.mkdtemp(suffix = 'aubio_slicing_test_case') |
---|
55 | |
---|
56 | def test_slice_start_empty(self): |
---|
57 | regions_start = [] |
---|
58 | self.assertRaises(ValueError, |
---|
59 | slice_source_at_stamps, |
---|
60 | self.source_file, regions_start, output_dir = self.output_dir) |
---|
61 | |
---|
62 | def test_slice_start_none(self): |
---|
63 | regions_start = None |
---|
64 | self.assertRaises(ValueError, |
---|
65 | slice_source_at_stamps, |
---|
66 | self.source_file, regions_start, output_dir = self.output_dir) |
---|
67 | |
---|
68 | def tearDown(self): |
---|
69 | shutil.rmtree(self.output_dir) |
---|
70 | |
---|
71 | class aubio_slicing_wrong_ends_test_case(TestCase): |
---|
72 | |
---|
73 | def setUp(self): |
---|
74 | self.source_file = get_default_test_sound(self) |
---|
75 | self.output_dir = tempfile.mkdtemp(suffix = 'aubio_slicing_test_case') |
---|
76 | |
---|
77 | def test_slice_wrong_ends(self): |
---|
78 | regions_start = [i*1000 for i in range(1, n_slices)] |
---|
79 | regions_end = [] |
---|
80 | self.assertRaises (ValueError, |
---|
81 | slice_source_at_stamps, self.source_file, regions_start, regions_end, |
---|
82 | output_dir = self.output_dir) |
---|
83 | |
---|
84 | def test_slice_no_ends(self): |
---|
85 | regions_start = [i*1000 for i in range(1, n_slices)] |
---|
86 | regions_end = None |
---|
87 | slice_source_at_stamps (self.source_file, regions_start, regions_end, |
---|
88 | output_dir = self.output_dir) |
---|
89 | original_samples = count_samples_in_file(self.source_file) |
---|
90 | written_samples = count_samples_in_directory(self.output_dir) |
---|
91 | assert_equal(written_samples, original_samples, |
---|
92 | "number of samples written different from number of original samples") |
---|
93 | |
---|
94 | def tearDown(self): |
---|
95 | shutil.rmtree(self.output_dir) |
---|
96 | |
---|
97 | if __name__ == '__main__': |
---|
98 | from unittest import main |
---|
99 | main() |
---|