Showing the changes to python/tests/test_source.py (1 of 2 files). Show all
Unified · Split
| @@ -32,7 +32,7 @@ class aubio_source_test_case(aubio_source_test_case_base): · 31 unmodified lines | |||
| 32 | 32 | ||
| 33 | class aubio_source_read_test_case(aubio_source_test_case_base): | 33 | class aubio_source_read_test_case(aubio_source_test_case_base): |
| 34 | 34 | ||
| 35 | def read_from_sink(self, f): | 35 | def read_from_source(self, f): |
| 36 | total_frames = 0 | 36 | total_frames = 0 |
| 37 | while True: | 37 | while True: |
| 38 | vec, read = f() | 38 | vec, read = f() |
| @@ -42,25 +42,26 @@ class aubio_source_read_test_case(aubio_source_test_case_base): · 3 unmodified lines | |||
| 42 | print "(", total_frames, "frames", "in", | 42 | print "(", total_frames, "frames", "in", |
| 43 | print total_frames / f.hop_size, "blocks", "at", "%dHz" % f.samplerate, ")", | 43 | print total_frames / f.hop_size, "blocks", "at", "%dHz" % f.samplerate, ")", |
| 44 | print "from", f.uri | 44 | print "from", f.uri |
| 45 | return total_frames | ||
| 45 | 46 | ||
| 46 | def test_samplerate_hopsize(self): | 47 | def test_samplerate_hopsize(self): |
| 47 | for p in list_of_sounds: | 48 | for p in list_of_sounds: |
| 48 | for samplerate, hop_size in zip([0, 44100, 8000, 32000], [ 512, 512, 64, 256]): | 49 | for samplerate, hop_size in zip([0, 44100, 8000, 32000], [ 512, 512, 64, 256]): |
| 49 | f = source(p, samplerate, hop_size) | 50 | f = source(p, samplerate, hop_size) |
| 50 | assert f.samplerate != 0 | 51 | assert f.samplerate != 0 |
| 51 | self.read_from_sink(f) | 52 | self.read_from_source(f) |
| 52 | 53 | ||
| 53 | def test_samplerate_none(self): | 54 | def test_samplerate_none(self): |
| 54 | for p in list_of_sounds: | 55 | for p in list_of_sounds: |
| 55 | f = source(p) | 56 | f = source(p) |
| 56 | assert f.samplerate != 0 | 57 | assert f.samplerate != 0 |
| 57 | self.read_from_sink(f) | 58 | self.read_from_source(f) |
| 58 | 59 | ||
| 59 | def test_samplerate_0(self): | 60 | def test_samplerate_0(self): |
| 60 | for p in list_of_sounds: | 61 | for p in list_of_sounds: |
| 61 | f = source(p, 0) | 62 | f = source(p, 0) |
| 62 | assert f.samplerate != 0 | 63 | assert f.samplerate != 0 |
| 63 | self.read_from_sink(f) | 64 | self.read_from_source(f) |
| 64 | 65 | ||
| 65 | def test_wrong_samplerate(self): | 66 | def test_wrong_samplerate(self): |
| 66 | for p in list_of_sounds: | 67 | for p in list_of_sounds: |
| @@ -85,11 +86,23 @@ class aubio_source_read_test_case(aubio_source_test_case_base): · 18 unmodified lines | |||
| 85 | f = source(p, 0, 0) | 86 | f = source(p, 0, 0) |
| 86 | assert f.samplerate != 0 | 87 | assert f.samplerate != 0 |
| 87 | assert f.hop_size != 0 | 88 | assert f.hop_size != 0 |
| 88 | self.read_from_sink(f) | 89 | self.read_from_source(f) |
| 90 | |||
| 91 | def test_seek_to_half(self): | ||
| 92 | from random import randint | ||
| 93 | for p in list_of_sounds: | ||
| 94 | f = source(p, 0, 0) | ||
| 95 | assert f.samplerate != 0 | ||
| 96 | assert f.hop_size != 0 | ||
| 97 | a = self.read_from_source(f) | ||
| 98 | c = randint(0, a) | ||
| 99 | f.seek(c) | ||
| 100 | b = self.read_from_source(f) | ||
| 101 | assert a == b + c | ||
| 89 | 102 | ||
| 90 | class aubio_source_readmulti_test_case(aubio_source_read_test_case): | 103 | class aubio_source_readmulti_test_case(aubio_source_read_test_case): |
| 91 | 104 | ||
| 92 | def read_from_sink(self, f): | 105 | def read_from_source(self, f): |
| 93 | total_frames = 0 | 106 | total_frames = 0 |
| 94 | while True: | 107 | while True: |
| 95 | vec, read = f.do_multi() | 108 | vec, read = f.do_multi() |
| @@ -100,6 +113,7 @@ class aubio_source_readmulti_test_case(aubio_source_read_test_case): · 4 unmodified lines | |||
| 100 | print f.channels, "channels and", | 113 | print f.channels, "channels and", |
| 101 | print total_frames / f.hop_size, "blocks", "at", "%dHz" % f.samplerate, ")", | 114 | print total_frames / f.hop_size, "blocks", "at", "%dHz" % f.samplerate, ")", |
| 102 | print "from", f.uri | 115 | print "from", f.uri |
| 116 | return total_frames | ||
| 103 | 117 | ||
| 104 | if __name__ == '__main__': | 118 | if __name__ == '__main__': |
| 105 | from unittest import main | 119 | from unittest import main |