Initial commit of OpenSPARC T2 architecture model.
[OpenSPARC-T2-SAM] / sam-t2 / devtools / v9 / lib / python2.4 / test / test_wave.py
CommitLineData
920dae64
AT
1from test.test_support import TestFailed, TESTFN
2import os
3import wave
4
5def check(t, msg=None):
6 if not t:
7 raise TestFailed, msg
8
9nchannels = 2
10sampwidth = 2
11framerate = 8000
12nframes = 100
13
14f = wave.open(TESTFN, 'wb')
15f.setnchannels(nchannels)
16f.setsampwidth(sampwidth)
17f.setframerate(framerate)
18f.setnframes(nframes)
19output = '\0' * nframes * nchannels * sampwidth
20f.writeframes(output)
21f.close()
22
23f = wave.open(TESTFN, 'rb')
24check(nchannels == f.getnchannels(), "nchannels")
25check(sampwidth == f.getsampwidth(), "sampwidth")
26check(framerate == f.getframerate(), "framerate")
27check(nframes == f.getnframes(), "nframes")
28input = f.readframes(nframes)
29check(input == output, "data")
30f.close()
31
32os.remove(TESTFN)