diff options
Diffstat (limited to 'numpy')
-rw-r--r-- | numpy/lib/tests/test_io.py | 33 | ||||
-rw-r--r-- | numpy/testing/tests/test_utils.py | 8 |
2 files changed, 18 insertions, 23 deletions
diff --git a/numpy/lib/tests/test_io.py b/numpy/lib/tests/test_io.py index bffc5c63e..45ee0a477 100644 --- a/numpy/lib/tests/test_io.py +++ b/numpy/lib/tests/test_io.py @@ -19,7 +19,7 @@ from numpy.ma.testutils import assert_equal from numpy.testing import ( TestCase, run_module_suite, assert_warns, assert_, assert_raises_regex, assert_raises, assert_allclose, - assert_array_equal, + assert_array_equal,temppath ) from numpy.testing.utils import tempdir @@ -259,26 +259,17 @@ class TestSavezLoad(RoundtripTest, TestCase): def test_not_closing_opened_fid(self): # Test that issue #2178 is fixed: # verify could seek on 'loaded' file - - fd, tmp = mkstemp(suffix='.npz') - os.close(fd) - try: - fp = open(tmp, 'wb') - np.savez(fp, data='LOVELY LOAD') - fp.close() - - fp = open(tmp, 'rb', 10000) - fp.seek(0) - assert_(not fp.closed) - np.load(fp)['data'] - # fp must not get closed by .load - assert_(not fp.closed) - fp.seek(0) - assert_(not fp.closed) - - finally: - fp.close() - os.remove(tmp) + with temppath(suffix='.npz') as tmp: + with open(tmp, 'wb') as fp: + np.savez(fp, data='LOVELY LOAD') + with open(tmp, 'rb', 10000) as fp: + fp.seek(0) + assert_(not fp.closed) + np.load(fp)['data'] + # fp must not get closed by .load + assert_(not fp.closed) + fp.seek(0) + assert_(not fp.closed) def test_closing_fid(self): # Test that issue #1517 (too many opened files) remains closed diff --git a/numpy/testing/tests/test_utils.py b/numpy/testing/tests/test_utils.py index b558270b3..23bd491bc 100644 --- a/numpy/testing/tests/test_utils.py +++ b/numpy/testing/tests/test_utils.py @@ -788,11 +788,13 @@ def test_tempdir(): pass assert_(not os.path.isdir(tdir)) + raised = False try: with tempdir() as tdir: raise ValueError() except ValueError: - pass + raised = True + assert_(raised) assert_(not os.path.isdir(tdir)) @@ -803,11 +805,13 @@ def test_temppath(): pass assert_(not os.path.isfile(fpath)) + raised = False try: with temppath() as fpath: raise ValueError() except ValueError: - pass + raised = True + assert_(raised) assert_(not os.path.isfile(fpath)) |