add tests for structured dtypes

Author: alimanfoo

zarr/tests/test_array.py

@@ -360,9 +360,11 @@ def _test_persistence(self, a, chunks):
         z3[:] = 0
 
         # check effect of write
-        assert_array_equal(np.zeros_like(a), z3[:])
-        assert_array_equal(np.zeros_like(a), z2[:])
-        assert_array_equal(np.zeros_like(a), z[:])
+        expect = np.empty_like(a)
+        expect[:] = 0
+        assert_array_equal(expect, z3[:])
+        assert_array_equal(expect, z2[:])
+        assert_array_equal(expect, z[:])
 
         # open for writing (must not exist)
         with assert_raises(ValueError):
@@ -372,13 +374,22 @@ def _test_persistence(self, a, chunks):
         shutil.rmtree(path)
 
     def test_persistence_1d(self):
+        # simple dtype
         self._test_persistence(np.arange(1050), chunks=(100,))
+        # structured dtype
+        dtype = np.dtype([('a', 'i4'), ('b', 'S10')])
+        self._test_persistence(np.empty(10000, dtype=dtype), chunks=(100,))
 
     def test_persistence_2d(self):
-        a = np.arange(10000).reshape((1000, 10))
         chunks = (100, 2)
+        # simple dtype
+        a = np.arange(10000).reshape((1000, 10))
         self._test_persistence(a, chunks=chunks)
-        
+        # structured dtype
+        dtype = np.dtype([('a', 'i4'), ('b', 'S10')])
+        self._test_persistence(np.empty((1000, 10), dtype=dtype),
+                               chunks=chunks)
+
     def test_resize_persistence(self):
 
         # setup path

zarr/tests/test_chunk.py

@@ -92,6 +92,14 @@ def test_create_chunk(self):
             self._test_create_chunk(np.linspace(-1, 1, 1e4, dtype=dtype)
                                     .reshape(100, -1))
 
+        # structured dtype
+        dtype = np.dtype([('a', 'i4'), ('b', 'S10')])
+        print(dtype)
+        print('1-dimensional')
+        self._test_create_chunk(np.empty(10000, dtype=dtype))
+        print('2-dimensional')
+        self._test_create_chunk(np.empty((100, 100), dtype=dtype))
+
     def test_create_chunk_fill_value(self):
 
         for shape in 100, (100, 100):