Guido van Rossum | 9a744a9 | 1999-04-08 20:27:54 +0000 | [diff] [blame] | 1 | |
| 2 | import sys, os |
| 3 | import gzip, tempfile |
| 4 | |
| 5 | filename = tempfile.mktemp() |
| 6 | |
| 7 | data1 = """ int length=DEFAULTALLOC, err = Z_OK; |
| 8 | PyObject *RetVal; |
| 9 | int flushmode = Z_FINISH; |
| 10 | unsigned long start_total_out; |
| 11 | |
| 12 | """ |
| 13 | |
| 14 | data2 = """/* zlibmodule.c -- gzip-compatible data compression */ |
| 15 | /* See http://www.cdrom.com/pub/infozip/zlib/ */ |
| 16 | /* See http://www.winimage.com/zLibDll for Windows */ |
| 17 | """ |
| 18 | |
| 19 | f = gzip.GzipFile(filename, 'wb') ; f.write(data1) ; f.close() |
| 20 | |
| 21 | f = gzip.GzipFile(filename, 'rb') ; d = f.read() ; f.close() |
| 22 | assert d == data1 |
| 23 | |
| 24 | # Append to the previous file |
| 25 | f = gzip.GzipFile(filename, 'ab') ; f.write(data2) ; f.close() |
| 26 | |
| 27 | f = gzip.GzipFile(filename, 'rb') ; d = f.read() ; f.close() |
| 28 | assert d == data1+data2 |
| 29 | |
| 30 | os.unlink( filename ) |