diff options
| author | Charles Harris <charlesr.harris@gmail.com> | 2016-05-09 12:37:01 -0600 |
|---|---|---|
| committer | Charles Harris <charlesr.harris@gmail.com> | 2016-05-09 12:37:01 -0600 |
| commit | cb6a5bfb12d87949a2cb18a801e09a0e59c05200 (patch) | |
| tree | e3e4326148e069ea7fd89eceb57aaf942acfb95b /numpy/lib/tests/test_format.py | |
| parent | 1b6bc8b2aa14bd228388568a95e6014578de77ad (diff) | |
| parent | da668fc74653e5caae9f741461c4d20f9df6e5c1 (diff) | |
| download | numpy-cb6a5bfb12d87949a2cb18a801e09a0e59c05200.tar.gz | |
Merge pull request #7598 from drasmuss/master
Cast array size to int64 when loading from archive
Diffstat (limited to 'numpy/lib/tests/test_format.py')
| -rw-r--r-- | numpy/lib/tests/test_format.py | 14 |
1 files changed, 14 insertions, 0 deletions
diff --git a/numpy/lib/tests/test_format.py b/numpy/lib/tests/test_format.py index a091ef5b3..46b21707f 100644 --- a/numpy/lib/tests/test_format.py +++ b/numpy/lib/tests/test_format.py @@ -836,5 +836,19 @@ def test_large_file_support(): assert_array_equal(r, d) +@dec.slow +def test_large_archive(): + a = np.empty((2 ** 30, 2), dtype=np.uint8) + fname = os.path.join(tempdir, "large_archive") + + with open(fname, "wb") as f: + np.savez(f, arr=a) + + with open(fname, "rb") as f: + new_a = np.load(f)["arr"] + + assert a.shape == new_a.shape + + if __name__ == "__main__": run_module_suite() |