| @@ -23,19 +23,27 @@ class aubio_slicing_test_case(TestCase): · 22 unmodified lines |
| 23 | | 23 | |
| 24 | def test_slice_start_only_no_zero(self): | 24 | def test_slice_start_only_no_zero(self): |
| 25 | regions_start = [i*1000 for i in range(1, n_slices)] | 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) | 26 | slice_source_at_stamps(self.source_file, regions_start, |
| | 27 | output_dir = self.output_dir, create_first=True) |
| 27 | | 28 | |
| 28 | def test_slice_start_beyond_end(self): | 29 | def test_slice_start_beyond_end(self): |
| 29 | regions_start = [i*1000 for i in range(1, n_slices)] | 30 | regions_start = [i*1000 for i in range(1, n_slices)] |
| 30 | regions_start += [count_samples_in_file(self.source_file) + 1000] | 31 | 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 | slice_source_at_stamps(self.source_file, regions_start, |
| | 33 | output_dir = self.output_dir, create_first=True) |
| 32 | | 34 | |
| 33 | def test_slice_start_every_blocksize(self): | 35 | def test_slice_start_every_blocksize(self): |
| 34 | hopsize = 200 | 36 | hopsize = 200 |
| 35 | regions_start = [i*hopsize for i in range(1, n_slices)] | 37 | regions_start = [i*hopsize for i in range(0, n_slices)] |
| 36 | slice_source_at_stamps(self.source_file, regions_start, output_dir = self.output_dir, | 38 | slice_source_at_stamps(self.source_file, regions_start, output_dir = self.output_dir, |
| 37 | hopsize = 200) | 39 | hopsize = 200) |
| 38 | | 40 | |
| | 41 | def test_slice_start_every_half_blocksize(self): |
| | 42 | hopsize = 200 |
| | 43 | regions_start = [i*hopsize//2 for i in range(0, n_slices)] |
| | 44 | slice_source_at_stamps(self.source_file, regions_start, |
| | 45 | output_dir = self.output_dir, hopsize = 200) |
| | 46 | |
| 39 | def tearDown(self): | 47 | def tearDown(self): |
| 40 | original_samples = count_samples_in_file(self.source_file) | 48 | original_samples = count_samples_in_file(self.source_file) |
| 41 | written_samples = count_samples_in_directory(self.output_dir) | 49 | written_samples = count_samples_in_directory(self.output_dir) |
| @@ -91,6 +99,19 @@ class aubio_slicing_with_ends_test_case(TestCase): · 49 unmodified lines |
| 91 | assert_equal(written_samples, expected_samples, | 99 | assert_equal(written_samples, expected_samples, |
| 92 | "number of samples written different from number of original samples") | 100 | "number of samples written different from number of original samples") |
| 93 | | 101 | |
| | 102 | def test_slice_start_and_ends_with_missing_end(self): |
| | 103 | regions_start = [i*1000 for i in range(n_slices)] |
| | 104 | regions_ends = [r-1 for r in regions_start[1:]] |
| | 105 | slice_source_at_stamps(self.source_file, regions_start, regions_ends, |
| | 106 | output_dir = self.output_dir) |
| | 107 | written_samples = count_samples_in_directory(self.output_dir) |
| | 108 | original_samples = count_samples_in_file(self.source_file) |
| | 109 | total_files = count_files_in_directory(self.output_dir) |
| | 110 | assert_equal(n_slices, total_files, |
| | 111 | "number of slices created different from expected") |
| | 112 | assert_equal(written_samples, original_samples, |
| | 113 | "number of samples written different from number of original samples") |
| | 114 | |
| 94 | def tearDown(self): | 115 | def tearDown(self): |
| 95 | shutil.rmtree(self.output_dir) | 116 | shutil.rmtree(self.output_dir) |
| 96 | | 117 | |
| @@ -133,7 +154,7 @@ class aubio_slicing_wrong_ends_test_case(TestCase): · 36 unmodified lines |
| 133 | regions_start = [i*1000 for i in range(1, n_slices)] | 154 | regions_start = [i*1000 for i in range(1, n_slices)] |
| 134 | regions_end = None | 155 | regions_end = None |
| 135 | slice_source_at_stamps (self.source_file, regions_start, regions_end, | 156 | slice_source_at_stamps (self.source_file, regions_start, regions_end, |
| 136 | output_dir = self.output_dir) | 157 | output_dir = self.output_dir, create_first=True) |
| 137 | total_files = count_files_in_directory(self.output_dir) | 158 | total_files = count_files_in_directory(self.output_dir) |
| 138 | assert_equal(n_slices, total_files, | 159 | assert_equal(n_slices, total_files, |
| 139 | "number of slices created different from expected") | 160 | "number of slices created different from expected") |