Skip to content

Commit 6f40926

Browse files
committed
ENH: Add large image sharding test
1 parent 853fdd2 commit 6f40926

File tree

1 file changed

+26
-0
lines changed

1 file changed

+26
-0
lines changed

test/test_large_serialization.py

Lines changed: 26 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -24,3 +24,29 @@ def test_large_image_serialization(input_images):
2424
# verify_against_baseline(dataset_name, baseline_name, multiscales)
2525

2626
config.memory_target = default_mem_target
27+
28+
29+
def test_large_image_serialization_with_sharding(input_images):
30+
default_mem_target = config.memory_target
31+
config.memory_target = int(1e6)
32+
33+
dataset_name = "lung_series"
34+
data = imread.imread(input_images[dataset_name])
35+
image = to_ngff_image(
36+
data=data,
37+
dims=("z", "y", "x"),
38+
scale={"z": 2.5, "y": 1.40625, "x": 1.40625},
39+
translation={"z": 332.5, "y": 360.0, "x": 0.0},
40+
name="LIDC2",
41+
)
42+
multiscales = to_multiscales(image)
43+
# baseline_name = "auto/memory_target_1e6.zarr"
44+
# store_new_multiscales(dataset_name, baseline_name, multiscales)
45+
test_store = MemoryStore()
46+
chunks_per_shard = 1
47+
to_ngff_zarr(
48+
test_store, multiscales, version="0.5", chunks_per_shard=chunks_per_shard
49+
)
50+
# verify_against_baseline(dataset_name, baseline_name, multiscales)
51+
52+
config.memory_target = default_mem_target

0 commit comments

Comments
 (0)