source: python/tests/test_slicing.py @ f36ecea

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

python/lib/aubio/slicing.py: use start and end stamps, make sure read > 0, improve tests

  • Property mode set to 100755
File size: 3.7 KB
Line 
1#! /usr/bin/env python
2
3from numpy.testing import TestCase, run_module_suite
4from numpy.testing import assert_equal, assert_almost_equal
5
6from aubio import slice_source_at_stamps
7from utils import *
8
9import tempfile
10import shutil
11
12n_slices = 8
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        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
50class 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
71class 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
97if __name__ == '__main__':
98    from unittest import main
99    main()
Note: See TracBrowser for help on using the repository browser.