ensure_wkw with compression (#109)

* ensure_wkw with compression * update reference data
scalableminds · Aug 2, 2019 · 62a8fd5 · 62a8fd5
1 parent a7c1498
commit 62a8fd5
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 1 deletion.
diff --git a/testdata/tiff_mag_2_reference.tar.gz b/testdata/tiff_mag_2_reference.tar.gz
diff --git a/wkcuber/downsampling.py b/wkcuber/downsampling.py
@@ -169,7 +169,14 @@ def downsample(
         )
     )
 
-    ensure_wkw(target_wkw_info)
+    with open_wkw(source_wkw_info) as source_wkw:
+        num_channels = source_wkw.header.num_channels
+        header_block_type = (
+            wkw.Header.BLOCK_TYPE_LZ4HC if compress else wkw.Header.BLOCK_TYPE_RAW
+        )
+        ensure_wkw(
+            target_wkw_info, block_type=header_block_type, num_channels=num_channels
+        )
 
     with get_executor_for_args(args) as executor:
         futures = []