summaryrefslogtreecommitdiffstats
path: root/test/option/stack-size.py
diff options
context:
space:
mode:
Diffstat (limited to 'test/option/stack-size.py')
-rw-r--r--test/option/stack-size.py6
1 files changed, 2 insertions, 4 deletions
diff --git a/test/option/stack-size.py b/test/option/stack-size.py
index febec5a..53faa9e 100644
--- a/test/option/stack-size.py
+++ b/test/option/stack-size.py
@@ -41,10 +41,8 @@ test.subdir('work1', 'work2')
test.write('build.py', r"""
import sys
-contents = open(sys.argv[2], 'rb').read()
-file = open(sys.argv[1], 'wb')
-file.write(contents)
-file.close()
+with open(sys.argv[1], 'wb') as f, open(sys.argv[2], 'rb') as infp:
+ f.write(infp.read())
""")