summaryrefslogtreecommitdiff
path: root/numpy/lib/tests/test_format.py
diff options
context:
space:
mode:
Diffstat (limited to 'numpy/lib/tests/test_format.py')
-rw-r--r--numpy/lib/tests/test_format.py14
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()