From cbcb397302334b35ebba03ebf7c11ec848d26a42 Mon Sep 17 00:00:00 2001
From: PetroFrolov <petro.frolov@gmail.com>
Date: Tue, 23 Mar 2021 15:15:26 +0300
Subject: [PATCH] fix test

---
 joblib/test/test_numpy_pickle.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/joblib/test/test_numpy_pickle.py b/joblib/test/test_numpy_pickle.py
index db130b1f4..a6f4123bc 100644
--- a/joblib/test/test_numpy_pickle.py
+++ b/joblib/test/test_numpy_pickle.py
@@ -634,6 +634,7 @@ def test_in_memory_persistence():
     for obj in objs:
         f = io.BytesIO()
         numpy_pickle.dump(obj, f)
+        f.seek(0)
         obj_reloaded = numpy_pickle.load(f)
         if isinstance(obj, np.ndarray):
             np.testing.assert_array_equal(obj_reloaded, obj)
@@ -678,6 +679,7 @@ def test_file_handle_persistence_in_memory_mmap():
     buf = io.BytesIO()
 
     numpy_pickle.dump(obj, buf)
+    buf.seek(0)
 
     with warns(UserWarning) as warninfo:
         numpy_pickle.load(buf, mmap_mode='r+')