1 from test.test_support import TestFailed, TESTFN

     2 import os

     3 import wave

     4 

     5 def check(t, msg=None):

     6     if not t:

     7         raise TestFailed, msg

     8 

     9 nchannels = 2

    10 sampwidth = 2

    11 framerate = 8000

    12 nframes = 100

    13 

    14 f = wave.open(TESTFN, 'wb')

    15 f.setnchannels(nchannels)

    16 f.setsampwidth(sampwidth)

    17 f.setframerate(framerate)

    18 f.setnframes(nframes)

    19 output = '\0' * nframes * nchannels * sampwidth

    20 f.writeframes(output)

    21 f.close()

    22 

    23 f = wave.open(TESTFN, 'rb')

    24 check(nchannels == f.getnchannels(), "nchannels")

    25 check(sampwidth == f.getsampwidth(), "sampwidth")

    26 check(framerate == f.getframerate(), "framerate")

    27 check(nframes == f.getnframes(), "nframes")

    28 input = f.readframes(nframes)

    29 check(input == output, "data")

    30 f.close()

    31 

    32 os.remove(TESTFN)