def test_adjust_cor(self): images = generate_images() m = CORInspectionDialogModel( images, 5, ScalarCoR(20), ReconstructionParameters('FBP_CUDA', 'ram-lak'), False) m.centre_value = 5 m.step = 1 m.adjust(ImageType.CURRENT) self.assertEqual(m.centre_value, 5) self.assertEqual(m.step, 0.5) m.adjust(ImageType.LESS) self.assertEqual(m.centre_value, 4.5) self.assertEqual(m.step, 0.5) m.adjust(ImageType.CURRENT) self.assertEqual(m.centre_value, 4.5) self.assertEqual(m.step, 0.25) m.adjust(ImageType.MORE) self.assertEqual(m.centre_value, 4.75) self.assertEqual(m.step, 0.25) m.adjust(ImageType.CURRENT) self.assertEqual(m.centre_value, 4.75) self.assertEqual(m.step, 0.125)
def test_iterations(self): images = generate_images() m = CORInspectionDialogModel( images, 5, ScalarCoR(20), ReconstructionParameters('FBP_CUDA', 'ram-lak'), True) m.adjust(ImageType.LESS) self.assertEqual(m.centre_value, 50) self.assertEqual(m.step, 50) m.adjust(ImageType.CURRENT) self.assertEqual(m.centre_value, 50) self.assertEqual(m.step, 25) m.adjust(ImageType.MORE) self.assertEqual(m.centre_value, 75) self.assertEqual(m.step, 25)