Ejemplo n.º 1
0
 def setUp(self):
     self.data = ImgData()
     self.data.load('Data/test_001.tif')
Ejemplo n.º 2
0
class XrsDataTest(unittest.TestCase):
    def setUp(self):
        self.data = ImgData()
        self.data.load('Data/test_001.tif')

    def test_rotation(self):
        pre_transformed_data = self.data.get_img_data()
        self.data.rotate_img_m90()
        self.data.rotate_img_m90()
        self.data.rotate_img_m90()
        self.data.rotate_img_m90()
        self.assertTrue(np.array_equal(self.data.get_img_data(), pre_transformed_data))
        self.data.reset_img_transformations()

        pre_transformed_data = self.data.get_img_data()
        self.data.rotate_img_m90()
        self.data.rotate_img_p90()
        self.assertTrue(np.array_equal(self.data.get_img_data(), pre_transformed_data))
        self.data.reset_img_transformations()

        pre_transformed_data = self.data.get_img_data()
        self.data.flip_img_horizontally()
        self.data.flip_img_horizontally()
        self.assertTrue(np.array_equal(self.data.get_img_data(), pre_transformed_data))
        self.data.reset_img_transformations()

        pre_transformed_data = self.data.get_img_data()
        self.data.flip_img_vertically()
        self.data.flip_img_vertically()
        self.assertTrue(np.array_equal(self.data.get_img_data(), pre_transformed_data))
        self.data.reset_img_transformations()

        self.data.flip_img_vertically()
        self.data.flip_img_horizontally()
        self.data.rotate_img_m90()
        self.data.rotate_img_p90()
        self.data.rotate_img_m90()
        self.data.rotate_img_m90()
        self.data.flip_img_horizontally()
        transformed_data = self.data.get_img_data()
        self.data.load('Data/test_001.tif')
        self.assertTrue(np.array_equal(self.data.get_img_data(), transformed_data))
        self.data.reset_img_transformations()

        pre_transformed_data = self.data.get_img_data()
        self.data.rotate_img_m90()
        self.data.reset_img_transformations()
        self.assertTrue(np.array_equal(self.data.get_img_data(), pre_transformed_data))

        pre_transformed_data = self.data.get_img_data()
        self.data.rotate_img_p90()
        self.data.reset_img_transformations()
        self.assertTrue(np.array_equal(self.data.get_img_data(), pre_transformed_data))

        pre_transformed_data = self.data.get_img_data()
        self.data.flip_img_horizontally()
        self.data.reset_img_transformations()
        self.assertTrue(np.array_equal(self.data.get_img_data(), pre_transformed_data))

        pre_transformed_data = self.data.get_img_data()
        self.data.flip_img_vertically()
        self.data.reset_img_transformations()
        self.assertTrue(np.array_equal(self.data.get_img_data(), pre_transformed_data))

        pre_transformed_data = self.data.get_img_data()
        self.data.flip_img_vertically()
        self.data.flip_img_horizontally()
        self.data.rotate_img_m90()
        self.data.rotate_img_p90()
        self.data.rotate_img_m90()
        self.data.rotate_img_m90()
        self.data.flip_img_horizontally()
        self.data.reset_img_transformations()
        self.assertTrue(np.array_equal(self.data.get_img_data(), pre_transformed_data))