Merge pull request #43 from pmarks/pmarks/allow-empty-chunked-resizable

allow the initial shape to be smaller than the chunk size, if the dataset is resizable

Author: aldanor

src/hl/dataset.rs

@@ -326,11 +326,14 @@ impl<T: H5Type> DatasetBuilder<T> {
                         "Invalid chunk: {:?} (all dimensions must be positive)",
                         dims
                     );
-                    ensure!(
-                        dims.iter().zip(shape.dims().iter()).all(|(&c, &s)| c <= s),
-                        "Invalid chunk: {:?} (must not exceed data shape in any dimension)",
-                        dims
-                    );
+
+                    if !self.resizable {
+                        ensure!(
+                            dims.iter().zip(shape.dims().iter()).all(|(&c, &s)| c <= s),
+                            "Invalid chunk: {:?} (must not exceed data shape in any dimension)",
+                            dims
+                        );
+                    }
 
                     let c_dims: Vec<hsize_t> = dims.iter().map(|&x| x as _).collect();
                     h5try!(H5Pset_chunk(id, dims.ndim() as _, c_dims.as_ptr()));
@@ -516,6 +519,19 @@ pub mod tests {
         })
     }
 
+    #[test]
+    pub fn test_chunks_resizable_zero_size() {
+        with_tmp_file(|file| {
+            let ds = file.new_dataset::<u32>().chunk((128,)).resizable(true).create("chunked_empty", (0,)).unwrap();
+            assert_eq!(ds.shape(), vec![0]);
+
+            ds.resize((10,)).unwrap();
+            assert_eq!(ds.shape(), vec![10]);
+
+            ds.as_writer().write(&vec![3; 10]).unwrap();
+        })
+    }
+
     #[test]
     pub fn test_invalid_chunk() {
         with_tmp_file(|file| {