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)