Python get_nx_tomo_test_data Examples

Example #1

File: data_util_test.py Project: imaluengo/Savu

 def test_get_slice_list_per_process(self):
     data = tu.get_nx_tomo_test_data()
     sl = du.get_grouped_slice_list(data, struct.CD_PROJECTION, 8)
     sl0 = du.get_slice_list_per_process(sl, 0, 4)
     sl1 = du.get_slice_list_per_process(sl, 1, 4)
     sl2 = du.get_slice_list_per_process(sl, 2, 4)
     sl3 = du.get_slice_list_per_process(sl, 3, 4)
     slt = sl0 + sl1 + sl2 + sl3
     self.assertListEqual(sl, slt)
     
     sl = du.get_grouped_slice_list(data, struct.CD_SINOGRAM, 8)
     sl0 = du.get_slice_list_per_process(sl, 0, 4)
     sl1 = du.get_slice_list_per_process(sl, 1, 4)
     sl2 = du.get_slice_list_per_process(sl, 2, 4)
     sl3 = du.get_slice_list_per_process(sl, 3, 4)
     slt = sl0 + sl1 + sl2 + sl3
     self.assertListEqual(sl, slt)
     
     sl = du.get_grouped_slice_list(data, struct.CD_ROTATION_AXIS, 8)
     sl0 = du.get_slice_list_per_process(sl, 0, 4)
     sl1 = du.get_slice_list_per_process(sl, 1, 4)
     sl2 = du.get_slice_list_per_process(sl, 2, 4)
     sl3 = du.get_slice_list_per_process(sl, 3, 4)
     slt = sl0 + sl1 + sl2 + sl3
     self.assertListEqual(sl, slt)

Example #2

File: data_util_test.py Project: yskashyap/Savu

    def test_get_slice_list_per_process(self):
        data = tu.get_nx_tomo_test_data()
        sl = du.get_grouped_slice_list(data, struct.CD_PROJECTION, 8)
        sl0 = du.get_slice_list_per_process(sl, 0, 4)
        sl1 = du.get_slice_list_per_process(sl, 1, 4)
        sl2 = du.get_slice_list_per_process(sl, 2, 4)
        sl3 = du.get_slice_list_per_process(sl, 3, 4)
        slt = sl0 + sl1 + sl2 + sl3
        self.assertListEqual(sl, slt)

        sl = du.get_grouped_slice_list(data, struct.CD_SINOGRAM, 8)
        sl0 = du.get_slice_list_per_process(sl, 0, 4)
        sl1 = du.get_slice_list_per_process(sl, 1, 4)
        sl2 = du.get_slice_list_per_process(sl, 2, 4)
        sl3 = du.get_slice_list_per_process(sl, 3, 4)
        slt = sl0 + sl1 + sl2 + sl3
        self.assertListEqual(sl, slt)

        sl = du.get_grouped_slice_list(data, struct.CD_ROTATION_AXIS, 8)
        sl0 = du.get_slice_list_per_process(sl, 0, 4)
        sl1 = du.get_slice_list_per_process(sl, 1, 4)
        sl2 = du.get_slice_list_per_process(sl, 2, 4)
        sl3 = du.get_slice_list_per_process(sl, 3, 4)
        slt = sl0 + sl1 + sl2 + sl3
        self.assertListEqual(sl, slt)

Example #3

File: data_util_test.py Project: yskashyap/Savu

 def test_slice_list(self):
     data = tu.get_nx_tomo_test_data()
     sl = du.get_slice_list(data, struct.CD_PROJECTION)
     self.assertEqual(len(sl), 111)
     sl = du.get_slice_list(data, struct.CD_SINOGRAM)
     self.assertEqual(len(sl), 135)
     sl = du.get_slice_list(data, struct.CD_ROTATION_AXIS)
     self.assertEqual(len(sl), 21600)

Example #4

File: data_util_test.py Project: imaluengo/Savu

 def test_slice_list(self):
     data = tu.get_nx_tomo_test_data()
     sl = du.get_slice_list(data, struct.CD_PROJECTION)
     self.assertEqual(len(sl), 111)
     sl = du.get_slice_list(data, struct.CD_SINOGRAM)
     self.assertEqual(len(sl), 135)
     sl = du.get_slice_list(data, struct.CD_ROTATION_AXIS)
     self.assertEqual(len(sl), 21600)

Example #5

File: data_util_test.py Project: imaluengo/Savu

    def test_get_padded_slice_data(self):
        data = tu.get_nx_tomo_test_data()
        sl = du.get_grouped_slice_list(data, struct.CD_PROJECTION, 8)
        
        padding_dict = {}
        padding_dict[struct.CD_ROTATION_AXIS] = 1

        psl0ra = du.get_padded_slice_data(sl[0], padding_dict, data)
        psl5ra = du.get_padded_slice_data(sl[5], padding_dict, data)
        self.assertEqual(psl0ra.shape, (10, 135, 160))
        self.assertEqual(psl5ra.shape, (10, 135, 160))

Example #6

File: data_util_test.py Project: yskashyap/Savu

    def test_get_padded_slice_data(self):
        data = tu.get_nx_tomo_test_data()
        sl = du.get_grouped_slice_list(data, struct.CD_PROJECTION, 8)

        padding_dict = {}
        padding_dict[struct.CD_ROTATION_AXIS] = 1

        psl0ra = du.get_padded_slice_data(sl[0], padding_dict, data)
        psl5ra = du.get_padded_slice_data(sl[5], padding_dict, data)
        self.assertEqual(psl0ra.shape, (10, 135, 160))
        self.assertEqual(psl5ra.shape, (10, 135, 160))

Example #7

File: data_util_test.py Project: yskashyap/Savu

    def test_get_padded_slice_data_2(self):
        data = tu.get_nx_tomo_test_data()
        sl = du.get_grouped_slice_list(data, struct.CD_ROTATION_AXIS, 8)

        padding_dict = {}
        padding_dict[struct.CD_ROTATION_AXIS] = 1

        psl0_0 = du.get_padded_slice_data(sl[0], padding_dict, data)
        psl5_0 = du.get_padded_slice_data(sl[5], padding_dict, data)
        self.assertEqual(psl0_0.shape, (113, 8))
        self.assertEqual(psl5_0.shape, (113, 8))

Example #8

File: data_util_test.py Project: imaluengo/Savu

 def test_get_padded_slice_data_2(self):
     data = tu.get_nx_tomo_test_data()
     sl = du.get_grouped_slice_list(data, struct.CD_ROTATION_AXIS, 8)
     
     padding_dict = {}
     padding_dict[struct.CD_ROTATION_AXIS] = 1
     
     psl0_0 = du.get_padded_slice_data(sl[0], padding_dict, data)
     psl5_0 = du.get_padded_slice_data(sl[5], padding_dict, data)
     self.assertEqual(psl0_0.shape, (113, 8))
     self.assertEqual(psl5_0.shape, (113, 8))

Example #9

File: test_util_test.py Project: yskashyap/Savu

 def testGetNXTomoExampleData(self):
     raw_data = tu.get_nx_tomo_test_data()
     self.assertEqual(raw_data.data.data.shape, (111, 135, 160),
                      "Data is not the correct shape, was " +
                      str(raw_data.data.data.shape))
     self.assertEqual(raw_data.control.data.shape, (111,),
                      "Control is not the correct shape, was " +
                      str(raw_data.control.data.shape))
     self.assertEqual(raw_data.image_key.data.shape, (111,),
                      "Image_key is not the correct shape, was " +
                      str(raw_data.image_key.data.shape))
     self.assertEqual(raw_data.rotation_angle.data.shape, (111,),
                      "Rotation_Angle is not the correct shape, was " +
                      str(raw_data.rotation_angle.data.shape))

Example #10

File: data_util_test.py Project: yskashyap/Savu

    def test_slice_group(self):
        data = tu.get_nx_tomo_test_data()
        sl = du.get_slice_list(data, struct.CD_PROJECTION)
        gsl = du.group_slice_list(sl, 8)
        self.assertEqual(len(gsl), 14)
        self.assertEqual(len(gsl[0]), 3)

        sl = du.get_slice_list(data, struct.CD_SINOGRAM)
        gsl = du.group_slice_list(sl, 8)
        self.assertEqual(len(gsl), 17)
        self.assertEqual(len(gsl[0]), 3)

        sl = du.get_slice_list(data, struct.CD_ROTATION_AXIS)
        gsl = du.group_slice_list(sl, 8)
        self.assertEqual(len(gsl), 2700)
        for group in gsl:
            self.assertEqual(len(group), 3)

Example #11

File: data_util_test.py Project: imaluengo/Savu

 def test_slice_group(self):
     data = tu.get_nx_tomo_test_data()
     sl = du.get_slice_list(data, struct.CD_PROJECTION)
     gsl = du.group_slice_list(sl, 8)
     self.assertEqual(len(gsl), 14)
     self.assertEqual(len(gsl[0]), 3)
     
     sl = du.get_slice_list(data, struct.CD_SINOGRAM)
     gsl = du.group_slice_list(sl, 8)
     self.assertEqual(len(gsl), 17)
     self.assertEqual(len(gsl[0]), 3)
     
     sl = du.get_slice_list(data, struct.CD_ROTATION_AXIS)
     gsl = du.group_slice_list(sl, 8)
     self.assertEqual(len(gsl), 2700)
     for group in gsl:
         self.assertEqual(len(group), 3)

Example #12

File: data_util_test.py Project: yskashyap/Savu

    def test_get_unpadded_slice_data_2(self):
        data = tu.get_nx_tomo_test_data()
        sl = du.get_grouped_slice_list(data, struct.CD_SINOGRAM, 1)

        padding_dict = {}
        #padding_dict[struct.CD_PROJECTION] = 1

        psl0_0 = du.get_padded_slice_data(sl[0], padding_dict, data)
        psl5_0 = du.get_padded_slice_data(sl[5], padding_dict, data)
        self.assertEqual(psl0_0.shape, (111, 1, 160))
        self.assertEqual(psl5_0.shape, (111, 1, 160))

        raw0 = data.data[sl[0]]
        psl0_un = du.get_unpadded_slice_data(sl[0], padding_dict, data, psl0_0)
        self.assertEqual(psl0_un.shape, raw0.shape)  # here
        self.assertEqual(psl0_un.sum(), raw0.sum())

        raw5 = data.data[sl[5]]
        psl5_un = du.get_unpadded_slice_data(sl[5], padding_dict, data, psl5_0)
        self.assertEqual(psl5_un.shape, raw5.shape)
        self.assertEqual(psl5_un.sum(), raw5.sum())

Example #13

File: data_util_test.py Project: yskashyap/Savu

    def test_get_unpadded_slice_data(self):
        data = tu.get_nx_tomo_test_data()
        sl = du.get_grouped_slice_list(data, struct.CD_ROTATION_AXIS, 8)

        padding_dict = {}
        padding_dict[struct.CD_PROJECTION] = 1

        psl0_0 = du.get_padded_slice_data(sl[0], padding_dict, data)
        psl5_0 = du.get_padded_slice_data(sl[5], padding_dict, data)
        self.assertEqual(psl0_0.shape, (111, 3, 10))
        self.assertEqual(psl5_0.shape, (111, 3, 10))

        raw0 = data.data[sl[0]]
        psl0_un = du.get_unpadded_slice_data(sl[0], padding_dict, data, psl0_0)
        self.assertEqual(psl0_un.squeeze().shape, raw0.shape)
        self.assertEqual(psl0_un.sum(), raw0.sum())

        raw5 = data.data[sl[5]]
        psl5_un = du.get_unpadded_slice_data(sl[5], padding_dict, data, psl5_0)
        self.assertEqual(psl5_un.squeeze().shape, raw5.shape)
        self.assertEqual(psl5_un.sum(), raw5.sum())

Example #14

File: data_util_test.py Project: imaluengo/Savu

 def test_get_unpadded_slice_data(self):
     data = tu.get_nx_tomo_test_data()
     sl = du.get_grouped_slice_list(data, struct.CD_ROTATION_AXIS, 8)
     
     padding_dict = {}
     padding_dict[struct.CD_PROJECTION] = 1
     
     psl0_0 = du.get_padded_slice_data(sl[0], padding_dict, data)
     psl5_0 = du.get_padded_slice_data(sl[5], padding_dict, data)
     self.assertEqual(psl0_0.shape, (111, 3, 10))
     self.assertEqual(psl5_0.shape, (111, 3, 10))
     
     raw0 = data.data[sl[0]]
     psl0_un = du.get_unpadded_slice_data(sl[0], padding_dict, data, psl0_0)
     self.assertEqual(psl0_un.squeeze().shape, raw0.shape)
     self.assertEqual(psl0_un.sum(), raw0.sum())
     
     raw5 = data.data[sl[5]]
     psl5_un = du.get_unpadded_slice_data(sl[5], padding_dict, data, psl5_0)
     self.assertEqual(psl5_un.squeeze().shape, raw5.shape)
     self.assertEqual(psl5_un.sum(), raw5.sum())

Example #15

File: data_util_test.py Project: imaluengo/Savu

 def test_get_unpadded_slice_data_2(self):
     data = tu.get_nx_tomo_test_data()
     sl = du.get_grouped_slice_list(data, struct.CD_SINOGRAM, 1)
     
     padding_dict = {}
     #padding_dict[struct.CD_PROJECTION] = 1
     
     psl0_0 = du.get_padded_slice_data(sl[0], padding_dict, data)
     psl5_0 = du.get_padded_slice_data(sl[5], padding_dict, data)
     self.assertEqual(psl0_0.shape, (111, 1, 160))
     self.assertEqual(psl5_0.shape, (111, 1, 160))
     
     raw0 = data.data[sl[0]]
     psl0_un = du.get_unpadded_slice_data(sl[0], padding_dict, data, psl0_0)
     self.assertEqual(psl0_un.shape, raw0.shape) # here
     self.assertEqual(psl0_un.sum(), raw0.sum())
     
     raw5 = data.data[sl[5]]
     psl5_un = du.get_unpadded_slice_data(sl[5], padding_dict, data, psl5_0)
     self.assertEqual(psl5_un.shape, raw5.shape)
     self.assertEqual(psl5_un.sum(), raw5.sum())

Example #16

File: structures_test.py Project: imaluengo/Savu

 def setUp(self):
     self.data = tu.get_nx_tomo_test_data()

Example #17

File: structures_test.py Project: yskashyap/Savu

 def setUp(self):
     self.data = tu.get_nx_tomo_test_data()

Example #18

File: structures_test.py Project: yskashyap/Savu

 def setUp(self):
     self.nx_tomo_data = tu.get_nx_tomo_test_data()
     self.projection_data = tu.get_projection_test_data()

Example #19

File: structures_test.py Project: imaluengo/Savu

 def setUp(self):
     self.nx_tomo_data = tu.get_nx_tomo_test_data()
     self.projection_data = tu.get_projection_test_data()

Example #20

File: data_test.py Project: mjn19172/Savu

 def test_create_smaller_data_block(self):
     data = tu.get_nx_tomo_test_data()
     plugin = pu.load_plugin("savu.plugins.downsample_filter")
     output = tu.get_appropriate_output_data(plugin, data)[0]
     self.assertEqual(output.get_data_shape(), (111, 68, 80))

Example #21

File: data_test.py Project: yskashyap/Savu

 def test_create_smaller_data_block(self):
     data = tu.get_nx_tomo_test_data()
     plugin = pu.load_plugin("savu.plugins.downsample_filter")
     output = tu.get_appropriate_output_data(plugin, data)[0]
     self.assertEqual(output.get_data_shape(), (111, 68, 80))