def test_3d_rgb(self):
testing_dir = os.path.join(os.path.dirname(os.path.realpath(__file__)), "testing_data")
keys = "image"
xforms = Compose(
[
LoadImaged(keys=keys),
AddChanneld(keys=keys),
ScaleIntensityd(keys=keys),
# change to RGB color image
RepeatChanneld(keys=keys, repeats=3),
]
)
image_path = os.path.join(testing_dir, "anatomical.nii")
ims = xforms({keys: image_path})
fig = pyplot.figure() # external figure
fig, _ = matshow3d(
volume=ims[keys],
fig=fig,
figsize=(2, 2),
frames_per_row=5,
every_n=2,
frame_dim=-1,
channel_dim=0,
show=False,
)
with tempfile.TemporaryDirectory() as tempdir:
tempimg = f"{tempdir}/matshow3d_rgb_test.png"
fig.savefig(tempimg)
comp = compare_images(f"{testing_dir}/matshow3d_rgb_test.png", tempimg, 5e-2)
self.assertIsNone(comp, f"value of comp={comp}") # None indicates test passed
def test_shape(self, input_param, input_data, expected_shape):
result = RepeatChanneld(**input_param)(input_data)
self.assertEqual(result["img"].shape, expected_shape)