| from test_support import TestFailed |
| import os, tempfile |
| import wave |
| |
| def check(t, msg=None): |
| if not t: |
| raise TestFailed, msg |
| |
| nchannels = 2 |
| sampwidth = 2 |
| framerate = 8000 |
| nframes = 100 |
| |
| testfile = tempfile.mktemp() |
| |
| f = wave.open(testfile, 'wb') |
| f.setnchannels(nchannels) |
| f.setsampwidth(sampwidth) |
| f.setframerate(framerate) |
| f.setnframes(nframes) |
| output = '\0' * nframes * nchannels * sampwidth |
| f.writeframes(output) |
| f.close() |
| |
| f = wave.open(testfile, 'rb') |
| check(nchannels == f.getnchannels(), "nchannels") |
| check(sampwidth == f.getsampwidth(), "sampwidth") |
| check(framerate == f.getframerate(), "framerate") |
| check(nframes == f.getnframes(), "nframes") |
| input = f.readframes(nframes) |
| check(input == output, "data") |
| f.close() |
| |
| os.remove(testfile) |