20 lines
678 B
Python
20 lines
678 B
Python
from contextlib import ExitStack
|
|
from itertools import product
|
|
from pathlib import Path
|
|
|
|
import numpy as np
|
|
from tqdm.auto import tqdm
|
|
|
|
from tiffwrite import IJTiffFile
|
|
|
|
|
|
def test_mult(tmp_path: Path) -> None:
|
|
shape = (2, 3, 5)
|
|
paths = [tmp_path / f'test{i}.tif' for i in range(6)]
|
|
with ExitStack() as stack:
|
|
tifs = [stack.enter_context(IJTiffFile(path, shape)) for path in paths] # noqa
|
|
for c, z, t in tqdm(product(range(shape[0]), range(shape[1]), range(shape[2])), total=np.prod(shape)): # noqa
|
|
for tif in tifs:
|
|
tif.save(np.random.randint(0, 255, (64, 64)), c, z, t)
|
|
assert all([path.exists() for path in paths])
|