def test_resulting_file(self): """Make sure UnBZip2 changes are pushed through to actual file.""" import tempfile import os (fd, fn) = tempfile.mkstemp() os.close(fd) try: f = open(fn, "w") f.write(gz_compress("hello world!")) f.close() f = UnGZip(open(fn, "r+")) f.read(6) f.write("Ausralia!") f.seek(-6, 1) f.write("tralia!") f.close() self.assertEquals(open(fn).read(), gz_compress("hello Australia!")) finally: os.unlink(fn)
def makeFile(self, contents, mode): s = StringIO(gz_compress(contents)) f = UnGZip(s, mode) f.getvalue = def_getvalue_maybe_buffered(f, s, gz_decompress) return f