def test_3d_multi(self):
data = np.random.rand(5, 8, 10, 12)
image = io.nib_to_sitk(data, self.affine)
assert image.GetDimension() == 3
assert image.GetSize() == (8, 10, 12)
assert image.GetNumberOfComponentsPerPixel() == 5
def test_2d_3d_multi(self):
data = np.random.rand(5, 10, 12, 1)
image = io.nib_to_sitk(data, self.affine, force_3d=True)
assert image.GetDimension() == 3
assert image.GetSize() == (10, 12, 1)
assert image.GetNumberOfComponentsPerPixel() == 5
def test_wrong_num_dims(self):
with self.assertRaises(ValueError):
io.nib_to_sitk(np.random.rand(10, 10), self.affine)
def test_2d_single(self):
data = np.random.rand(1, 10, 12, 1)
image = io.nib_to_sitk(data, self.affine)
assert image.GetDimension() == 2
assert image.GetSize() == (10, 12)
assert image.GetNumberOfComponentsPerPixel() == 1
def nib_to_sitk(data: TypeData, affine: TypeData):
return nib_to_sitk(data, affine)
