def _make_image(ignore):
if ignore:
pytest.skip()
direct_img = DirectImage(func=lambda: np.array(np.ones((10, 10))),
name="direct",
labels={"detectors"})
return [direct_img]
def _make_image_list(ignore):
if ignore:
pytest.skip()
direct_img_list = DirectImage(func=lambda: [[1] * 10] * 10,
name="direct",
labels={"detectors"})
direct_img_list.img.name = "direct_img_list"
return [direct_img_list]
def test_4d_data(RE, tmp_path):
"""
Expect 0 files written.
"""
def return_4d_data():
return np.zeros((3, 3, 3, 3))
det = DirectImage(name="4d_image", func=return_4d_data)
tiff_serializer = Serializer(directory=tmp_path)
RE.subscribe(tiff_serializer)
RE(count([det], num=5))
assert "stream_data" not in tiff_serializer.artifacts
filenames = os.listdir(path=tmp_path)
assert len(filenames) == 0
def test_3d_data(RE, tmp_path):
"""
Expect 3 files written for each detector trigger.
"""
def return_3d_data():
return np.zeros((3, 3, 3))
det = DirectImage(name="3d_image", func=return_3d_data)
tiff_serializer = Serializer(directory=tmp_path)
RE.subscribe(tiff_serializer)
RE(count([det], num=5))
assert len(tiff_serializer.artifacts["stream_data"]) == 15
filenames = os.listdir(path=tmp_path)
assert len(filenames) == 15
assert all([re.search(r"-\d+\.tiff$", filename) for filename in filenames])