Exemple #1
0
def test_theverybasics():
    nim = NiftiImage(image_file)
    ok_(nim.getDataArray().shape == data_shape)

    # basic incomplete property check
    ok_(nim.extent == (128, 96, 24, 2))
    ok_(nim.rtime == 2000)
    ok_((nim.pixdim == (2.0, 2.0, 2.1999990940093994, 2000.0, 1.0, 1.0, 1.0)).all())
    ok_(nim.voxdim == (2.0, 2.0, 2.1999990940093994))
    ok_(nim.header["datatype"] == 4)
    ok_(nim.header["bitpix"] == 16)
    ok_(nim.header["magic"] == "n+1")
Exemple #2
0
class FileIOTests(unittest.TestCase):
    def setUp(self):
        self.workdir = tempfile.mkdtemp()
        self.nimg = NiftiImage(image_file)
        self.fp = tempfile.NamedTemporaryFile(suffix='.nii.gz')
        self.fp_plain = tempfile.NamedTemporaryFile(suffix='.nii')


    def tearDown(self):
        shutil.rmtree(self.workdir)
        del self.nimg
        self.fp.close()
        self.fp_plain.close()


    def testIdempotentLoadSaveCycle(self):
        """ check if file is unchanged by load/save cycle.
        """
        md5_orig = md5sum(image_file)
        self.nimg.save(self.fp.name)
        self.nimg.save('failed.nii.gz')
        md5_io =  md5sum(self.fp.name)

        # ASK MATTHEW: Fails since all extensions get stripped
        #self.failUnlessEqual(md5_orig, md5_io)


#    def testUnicodeLoadSaveCycle(self):
#        """ check load/save cycle for unicode filenames.
#        """
#        md5_orig = md5sum(os.path.join(example_data_path, 'example4d.nii.gz'))
#        self.nimg.save( os.path.join( self.workdir, 'üöä.nii.gz') )
#        md5_io =  md5sum( os.path.join( self.workdir, 'üöä.nii.gz') )
#
#        self.failUnlessEqual(md5_orig, md5_io)
#
#
    def testDataAccess(self):
        # test two points
        self.failUnlessEqual(self.nimg.data[1,12,59,49], 509)
        self.failUnlessEqual(self.nimg.data[0,4,17,42], 435)
Exemple #3
0
 def setUp(self):
     self.workdir = tempfile.mkdtemp()
     self.nimg = NiftiImage(image_file)
     self.fp = tempfile.NamedTemporaryFile(suffix='.nii.gz')
     self.fp_plain = tempfile.NamedTemporaryFile(suffix='.nii')