def setUp(self): self.data = ImgData() self.data.load('Data/test_001.tif')
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))