1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
|
# Test deflate.DeflateIO compression, with out-of-memory errors.
try:
# Check if deflate is available.
import deflate
import io
except ImportError:
print("SKIP")
raise SystemExit
# Check if compression is enabled.
if not hasattr(deflate.DeflateIO, "write"):
print("SKIP")
raise SystemExit
# Create a compressor object.
b = io.BytesIO()
g = deflate.DeflateIO(b, deflate.RAW, 15)
# Then, use up most of the heap.
l = []
while True:
try:
l.append(bytearray(1000))
except:
break
l.pop()
# Try to compress. This will try to allocate a large window and fail.
try:
g.write("test")
except MemoryError:
print("MemoryError")
# Should still be able to close the stream.
g.close()
# The underlying output stream should be unchanged.
print(b.getvalue())
|
