Added some more tests just in case.

tests/test_delayed_array.py

@@ -115,7 +115,7 @@ def test_delayed_array_low_block_size_F_contiguous():
 ########################################################
 
 
-def test_delayed_array_sparse():
+def test_delayed_array_sparse_csc():
     x = scipy.sparse.random(1000, 200, 0.1).tocsc()
     y = da.wrap(x) * 10
 
@@ -127,3 +127,17 @@ def test_delayed_array_sparse():
     assert isinstance(roundtrip, dm.ReloadedArray)
     assert isinstance(roundtrip.seed.seed, filebackedarray.Hdf5CompressedSparseMatrixSeed)
     assert (numpy.array(roundtrip) == x.toarray() * 10).all()
+
+
+def test_delayed_array_sparse_csr():
+    x = scipy.sparse.random(1000, 200, 0.1).tocsr()
+    y = da.wrap(x) * 10
+
+    dir = os.path.join(mkdtemp(), "foobar")
+    dl.save_object(y, dir, compressed_sparse_matrix_buffer_size=8*10000)
+    roundtrip = dl.read_object(dir)
+    assert roundtrip.shape == y.shape
+    assert roundtrip.dtype == y.dtype
+    assert isinstance(roundtrip, dm.ReloadedArray)
+    assert isinstance(roundtrip.seed.seed, filebackedarray.Hdf5CompressedSparseMatrixSeed)
+    assert (numpy.array(roundtrip) == x.toarray() * 10).all()

tests/test_optimize_storage.py

@@ -177,12 +177,12 @@ def test_optimize_integer_storage_scipy():
     assert opt.placeholder is None
     assert opt.non_zero == 5
 
-    opt = optim.optimize_integer_storage(y.tocsc())
+    opt = optim.optimize_integer_storage(y.tocsc(), buffer_size = 10)
     assert opt.type == "i2"
     assert opt.placeholder is None
     assert opt.non_zero == 5
 
-    opt = optim.optimize_integer_storage(y.tocsr())
+    opt = optim.optimize_integer_storage(y.tocsr(), buffer_size = 20)
     assert opt.type == "i2"
     assert opt.placeholder is None
     assert opt.non_zero == 5
@@ -450,12 +450,12 @@ def test_optimize_float_storage_scipy():
     assert opt.placeholder is None
     assert opt.non_zero == 5
 
-    opt = optim.optimize_float_storage(y.tocsc())
+    opt = optim.optimize_float_storage(y.tocsc(), buffer_size = 10)
     assert opt.type == "i2"
     assert opt.placeholder is None
     assert opt.non_zero == 5
 
-    opt = optim.optimize_float_storage(y.tocsr())
+    opt = optim.optimize_float_storage(y.tocsr(), buffer_size = 20)
     assert opt.type == "i2"
     assert opt.placeholder is None
     assert opt.non_zero == 5
@@ -629,12 +629,12 @@ def test_optimize_boolean_storage_scipy():
     assert opt.placeholder is None
     assert opt.non_zero == 5
 
-    opt = optim.optimize_boolean_storage(y.tocsc())
+    opt = optim.optimize_boolean_storage(y.tocsc(), buffer_size = 2)
     assert opt.type == "i1"
     assert opt.placeholder is None
     assert opt.non_zero == 5
 
-    opt = optim.optimize_boolean_storage(y.tocsr())
+    opt = optim.optimize_boolean_storage(y.tocsr(), buffer_size = 5)
     assert opt.type == "i1"
     assert opt.placeholder is None
     assert opt.non_zero == 5