Skip to content

Commit 1a11d0d

Browse files
committed
Add initial testing for zarr converter
1 parent 0f1222c commit 1a11d0d

File tree

1 file changed

+33
-0
lines changed

1 file changed

+33
-0
lines changed

test/test_zarr_converter.py

Lines changed: 33 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,33 @@
1+
from click.testing import CliRunner
2+
import pytools.zarr_converter
3+
import pytest
4+
import SimpleITK as sitk
5+
import json
6+
from pathlib import Path
7+
8+
args = ["--help", "--version"]
9+
10+
11+
@pytest.mark.parametrize("cli_args", args)
12+
def test_zarr_converter_main_help(cli_args):
13+
runner = CliRunner()
14+
result = runner.invoke(pytools.zarr_converter.main, cli_args.split())
15+
assert not result.exception
16+
17+
18+
@pytest.mark.parametrize(
19+
"image_mrc",
20+
[sitk.sitkUInt8, sitk.sitkInt16, sitk.sitkUInt16, sitk.sitkFloat32],
21+
indirect=["image_mrc"],
22+
)
23+
def test_zarr_converter1(image_mrc):
24+
runner = CliRunner()
25+
zarr_path = "output.zarr"
26+
with runner.isolated_filesystem():
27+
result = runner.invoke(pytools.zarr_converter.main, [image_mrc, zarr_path])
28+
assert not result.exception
29+
30+
with open(Path(zarr_path) / ".zattrs") as fp:
31+
zattrs = json.load(fp)
32+
assert "multiscales" in zattrs
33+
assert "datasets" in zattrs["multiscales"][0]

0 commit comments

Comments
 (0)