Beispiel #1
0
    def test_get_methods(self):

        tmp_arr = np.ones((5, 5), dtype=np.uint8)
        vox = [1.0, 1.0]
        orig = [0, 0]
        img = SpatialImage(tmp_arr)
        metadata = {
            'dim': 2,
            'extent': [5.0, 5.0],
            'shape': (5, 5),
            'type': 'uint8',
            'voxelsize': [1.0, 1.0],
            'origin': [0, 0],
            'max': 1,
            'mean': 1.0,
            'min': 1
        }

        self.assertEqual(img.get_type(), tmp_arr.dtype)
        self.assertEqual(img.get_voxelsize(), vox)
        self.assertEqual(img.get_shape(), tmp_arr.shape)
        self.assertEqual(img.get_origin(), orig)
        self.assertEqual(img.get_extent(),
                         [tmp_arr.shape[0], tmp_arr.shape[1]])
        self.assertEqual(img.get_dim(), 2)
        self.assertDictEqual(img.get_metadata(), metadata)
    def test_get_methods(self):

        tmp_arr = np.ones((5,5),dtype=np.uint8)
        vox = [1.0, 1.0]
        orig = [0, 0]
        img = SpatialImage(tmp_arr)
        metadata = {'dim': 2, 'extent': [5.0, 5.0], 'shape': (5, 5), 'type': 'uint8',
                    'voxelsize': [1.0, 1.0], 'origin': [0, 0], 'max': 1, 'mean': 1.0,
                    'min': 1}

        self.assertEqual(img.get_type(), tmp_arr.dtype)
        self.assertEqual(img.get_voxelsize(), vox)
        self.assertEqual(img.get_shape(), tmp_arr.shape)
        self.assertEqual(img.get_origin(), orig)
        self.assertEqual(img.get_extent(), [tmp_arr.shape[0], tmp_arr.shape[1]])
        self.assertEqual(img.get_dim(), 2)
        self.assertDictEqual(img.get_metadata(), metadata)
Beispiel #3
0
    def test_set_methods(self):

        tmp_arr = np.ones((5, 5), dtype=np.uint8)
        img = SpatialImage(tmp_arr)
        new_vox = [0.5, 0.5]
        new_orig = [1, 1]
        new_ext = [
            tmp_arr.shape[0] * new_vox[0], tmp_arr.shape[1] * new_vox[1]
        ]
        new_type = np.uint16
        new_met = img.get_metadata()
        new_met['name'] = 'img_test'

        img.set_voxelsize(new_vox)
        self.assertEqual(img.get_voxelsize(), new_vox)
        self.assertEqual(img.get_metadata()['voxelsize'], new_vox)
        self.assertEqual(img.get_extent(), new_ext)
        self.assertEqual(img.get_metadata()['extent'], new_ext)
        img = img.set_type(new_type)
        self.assertEqual(img.get_type(), 'uint16')
        self.assertEqual(img.get_metadata()['type'], 'uint16')
        img.set_origin(new_orig)
        self.assertEqual(img.get_origin(), new_orig)
        self.assertEqual(img.get_metadata()['origin'], new_orig)
        img.set_extent([10.0, 5.0])
        self.assertEqual(img.get_voxelsize(), [2.0, 1.0])
        self.assertEqual(img.get_metadata()['extent'], [10.0, 5.0])
        self.assertEqual(img.get_metadata()['voxelsize'], [2.0, 1.0])
        img.set_metadata(new_met)
        self.assertDictEqual(img.get_metadata(), new_met)
        img.set_pixel([2, 2], 10)
        self.assertEqual(img.get_pixel([2, 2]), 10)

        #--- numpy compatibility (for example transposition)
        new_arr = np.ones((10, 5), dtype=np.uint8)
        img = SpatialImage(new_arr, voxelsize=[0.5, 1])
        img = img.transpose()
        self.assertEqual(img.get_metadata()['shape'], (5, 10))
        self.assertEqual(img.get_metadata()['voxelsize'], [1.0, 0.5])
        self.assertEqual(img.get_metadata()['voxelsize'], img.get_voxelsize())
        self.assertEqual(img.get_metadata()['extent'], [5.0, 5.0])
        self.assertEqual(img.get_metadata()['extent'], img.get_extent())
    def test_set_methods(self):

        tmp_arr = np.ones((5,5),dtype=np.uint8)
        img = SpatialImage(tmp_arr)
        new_vox = [0.5, 0.5]
        new_orig = [1, 1]
        new_ext = [tmp_arr.shape[0]*new_vox[0], tmp_arr.shape[1]*new_vox[1]]
        new_type = np.uint16
        new_met = img.get_metadata()
        new_met['name'] = 'img_test'

        img.set_voxelsize(new_vox)
        self.assertEqual(img.get_voxelsize(), new_vox)
        self.assertEqual(img.get_metadata()['voxelsize'], new_vox)
        self.assertEqual(img.get_extent(), new_ext)
        self.assertEqual(img.get_metadata()['extent'], new_ext)
        img = img.set_type(new_type)
        self.assertEqual(img.get_type(), 'uint16')
        self.assertEqual(img.get_metadata()['type'], 'uint16')
        img.set_origin(new_orig)
        self.assertEqual(img.get_origin(), new_orig)
        self.assertEqual(img.get_metadata()['origin'], new_orig)
        img.set_extent([10.0, 5.0])
        self.assertEqual(img.get_voxelsize(), [2.0,1.0])
        self.assertEqual(img.get_metadata()['extent'], [10.0, 5.0])
        self.assertEqual(img.get_metadata()['voxelsize'], [2.0,1.0])
        img.set_metadata(new_met)
        self.assertDictEqual(img.get_metadata(), new_met)
        img.set_pixel([2,2], 10)
        self.assertEqual(img.get_pixel([2,2]), 10)

        #--- numpy compatibility (for example transposition)
        new_arr = np.ones((10,5), dtype=np.uint8)
        img = SpatialImage(new_arr, voxelsize=[0.5,1])
        img = img.transpose()
        self.assertEqual(img.get_metadata()['shape'], (5,10))
        self.assertEqual(img.get_metadata()['voxelsize'], [1.0,0.5])
        self.assertEqual(img.get_metadata()['voxelsize'], img.get_voxelsize())
        self.assertEqual(img.get_metadata()['extent'], [5.0,5.0])
        self.assertEqual(img.get_metadata()['extent'], img.get_extent())