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