Lib/test/test_file.py

   1 import os
   2
   3 from test_support import verify, TESTFN
   4 from UserList import UserList
   5
   6 # verify writelines with instance sequence
   7 l = UserList(['1', '2'])
   8 f = open(TESTFN, 'wb')
   9 f.writelines(l)
  10 f.close()
  11 f = open(TESTFN, 'rb')
  12 buf = f.read()
  13 f.close()
  14 verify(buf == '12')
  15
  16 # verify writelines with integers
  17 f = open(TESTFN, 'wb')
  18 try:
  19     f.writelines([1, 2, 3])
  20 except TypeError:
  21     pass
  22 else:
  23     print "writelines accepted sequence of integers"
  24 f.close()
  25
  26 # verify writelines with integers in UserList
  27 f = open(TESTFN, 'wb')
  28 l = UserList([1,2,3])
  29 try:
  30     f.writelines(l)
  31 except TypeError:
  32     pass
  33 else:
  34     print "writelines accepted sequence of integers"
  35 f.close()
  36
  37 # verify writelines with non-string object
  38 class NonString: pass
  39
  40 f = open(TESTFN, 'wb')
  41 try:
  42     f.writelines([NonString(), NonString()])
  43 except TypeError:
  44     pass
  45 else:
  46     print "writelines accepted sequence of non-string objects"
  47 f.close()
  48
  49 os.unlink(TESTFN)