def test_array_2D(self): maps = Maps(self.array2D, Ni=2, Nj=1, Nk=1, affine=self.affine) img = maps.to_img() self.assertTrue( np.array_equal(img.get_fdata(), self.array2D.reshape( (2, 1, 1, 2)))) self.assertTrue(np.array_equal(img.affine, self.affine))
def test_array_2D_one(self): maps = Maps(self.array2D, Ni=2, Nj=1, Nk=1, affine=self.affine) img0 = maps.to_img(0) img1 = maps.to_img(1) self.assertTrue( np.array_equal(img0.get_fdata(), self.array2D.reshape((2, 1, 1, 2))[:, :, :, 0])) self.assertTrue( np.array_equal(img1.get_fdata(), self.array2D.reshape((2, 1, 1, 2))[:, :, :, 1])) self.assertTrue(np.array_equal(img0.affine, self.affine)) self.assertTrue(np.array_equal(img1.affine, self.affine))
def test_array_4D_one(self): maps = Maps(self.array4D, affine=self.affine) img = maps.to_img(0) self.assertTrue( np.array_equal(img.get_fdata(), self.array4D[:, :, :, 0])) self.assertTrue(np.array_equal(img.affine, self.affine))
def test_array_3D(self): maps = Maps(self.array3D, affine=self.affine) img = maps.to_img() self.assertTrue(np.array_equal(img.get_fdata(), self.array3D)) self.assertTrue(np.array_equal(img.affine, self.affine))
def test_forbidden_array_4D_2_one(self): maps = Maps(self.array4D_2) with self.assertRaises(ValueError): maps.to_img()
def test_forbidden_array_3D(self): maps = Maps(self.array3D) with self.assertRaises(ValueError): maps.to_img()
def test_forbidden_array_2D_one(self): maps = Maps(self.array2D, Ni=2, Nj=1, Nk=1) with self.assertRaises(ValueError): maps.to_img()
def test_loaded_4D_img(self): maps = Maps(fmri_img) self.assertTrue( np.array_equal(maps.to_img().get_fdata(), fmri_img.get_fdata()))
def test_loaded_3D_img(self): maps = Maps(template) self.assertTrue( np.array_equal(maps.to_img().get_fdata(), template.get_fdata()))
def test_array_4D_2_all(self): maps = Maps(self.array4D_2, affine=self.affine) img = maps.to_img() self.assertTrue(np.array_equal(img.get_fdata(), self.array4D_2)) self.assertTrue(np.array_equal(img.affine, self.affine))