diff options
Diffstat (limited to 'lib/python2.7/test/test_wave.py')
-rw-r--r-- | lib/python2.7/test/test_wave.py | 45 |
1 files changed, 45 insertions, 0 deletions
diff --git a/lib/python2.7/test/test_wave.py b/lib/python2.7/test/test_wave.py new file mode 100644 index 0000000..02f5fae --- /dev/null +++ b/lib/python2.7/test/test_wave.py @@ -0,0 +1,45 @@ +from test.test_support import TESTFN, run_unittest +import os +import wave +import unittest + +nchannels = 2 +sampwidth = 2 +framerate = 8000 +nframes = 100 + +class TestWave(unittest.TestCase): + + def setUp(self): + self.f = None + + def tearDown(self): + if self.f is not None: + self.f.close() + try: + os.remove(TESTFN) + except OSError: + pass + + def test_it(self): + self.f = wave.open(TESTFN, 'wb') + self.f.setnchannels(nchannels) + self.f.setsampwidth(sampwidth) + self.f.setframerate(framerate) + self.f.setnframes(nframes) + output = '\0' * nframes * nchannels * sampwidth + self.f.writeframes(output) + self.f.close() + + self.f = wave.open(TESTFN, 'rb') + self.assertEqual(nchannels, self.f.getnchannels()) + self.assertEqual(sampwidth, self.f.getsampwidth()) + self.assertEqual(framerate, self.f.getframerate()) + self.assertEqual(nframes, self.f.getnframes()) + self.assertEqual(self.f.readframes(nframes), output) + +def test_main(): + run_unittest(TestWave) + +if __name__ == '__main__': + test_main() |