def test_2d_3d_multi(self): data = np.random.rand(5, 10, 12, 1) image = 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_3d_single(self): data = np.random.rand(1, 8, 10, 12) image = nib_to_sitk(data, self.affine) assert image.GetDimension() == 3 assert image.GetSize() == (8, 10, 12) assert image.GetNumberOfComponentsPerPixel() == 1
def test_wrong_num_dims(self): with self.assertRaises(ValueError): nib_to_sitk(np.random.rand(10, 10), self.affine)