diff options
Diffstat (limited to 'pyximport/test')
| -rw-r--r-- | pyximport/test/test_pyximport.py | 29 | ||||
| -rw-r--r-- | pyximport/test/test_reload.py | 10 |
2 files changed, 22 insertions, 17 deletions
diff --git a/pyximport/test/test_pyximport.py b/pyximport/test/test_pyximport.py index b3a4a9058..b7fd8d7ee 100644 --- a/pyximport/test/test_pyximport.py +++ b/pyximport/test/test_pyximport.py @@ -41,34 +41,37 @@ def test_with_reload(): tempdir = make_tempdir() sys.path.append(tempdir) filename = os.path.join(tempdir, "dummy.pyx") - open(filename, "w").write("print 'Hello world from the Pyrex install hook'") + with open(filename, "w") as fid: + fid.write("print 'Hello world from the Pyrex install hook'") import dummy reload(dummy) depend_filename = os.path.join(tempdir, "dummy.pyxdep") - depend_file = open(depend_filename, "w") - depend_file.write("*.txt\nfoo.bar") - depend_file.close() + with open(depend_filename, "w") as depend_file: + depend_file.write("*.txt\nfoo.bar") build_filename = os.path.join(tempdir, "dummy.pyxbld") - build_file = open(build_filename, "w") - build_file.write(""" + with open(build_filename, "w") as build_file: + build_file.write(""" from distutils.extension import Extension def make_ext(name, filename): return Extension(name=name, sources=[filename]) """) - build_file.close() - open(os.path.join(tempdir, "foo.bar"), "w").write(" ") - open(os.path.join(tempdir, "1.txt"), "w").write(" ") - open(os.path.join(tempdir, "abc.txt"), "w").write(" ") + with open(os.path.join(tempdir, "foo.bar"), "w") as fid: + fid.write(" ") + with open(os.path.join(tempdir, "1.txt"), "w") as fid: + fid.write(" ") + with open(os.path.join(tempdir, "abc.txt"), "w") as fid: + fid.write(" ") reload(dummy) assert len(pyximport._test_files)==1, pyximport._test_files reload(dummy) - time.sleep(1) # sleep a second to get safer mtimes - open(os.path.join(tempdir, "abc.txt"), "w").write(" ") - print("Here goes the reolad") + time.sleep(1) # sleep a second to get safer mtimes + with open(os.path.join(tempdir, "abc.txt"), "w") as fid: + fid.write(" ") + print("Here goes the reload") reload(dummy) assert len(pyximport._test_files) == 1, pyximport._test_files diff --git a/pyximport/test/test_reload.py b/pyximport/test/test_reload.py index ba53746f9..0ba5ba13f 100644 --- a/pyximport/test/test_reload.py +++ b/pyximport/test/test_reload.py @@ -18,14 +18,16 @@ def test(): tempdir = test_pyximport.make_tempdir() sys.path.append(tempdir) hello_file = os.path.join(tempdir, "hello.pyx") - open(hello_file, "w").write("x = 1; print x; before = 'before'\n") + with open(hello_file, "w") as fid: + fid.write("x = 1; print x; before = 'before'\n") import hello assert hello.x == 1 - time.sleep(1) # sleep to make sure that new "hello.pyx" has later - # timestamp than object file. + time.sleep(1) # sleep to make sure that new "hello.pyx" has later + # timestamp than object file. - open(hello_file, "w").write("x = 2; print x; after = 'after'\n") + with open(hello_file, "w") as fid: + fid.write("x = 2; print x; after = 'after'\n") reload(hello) assert hello.x == 2, "Reload should work on Python 2.3 but not 2.2" test_pyximport.remove_tempdir(tempdir) |