Skip to content

Commit

Permalink
Test fixes
Browse files Browse the repository at this point in the history
  • Loading branch information
DolicaAkelloEgwel committed Nov 30, 2020
1 parent 8eb392a commit 89232d0
Showing 1 changed file with 4 additions and 4 deletions.
8 changes: 4 additions & 4 deletions mantidimaging/gui/dialogs/cor_inspection/test/model_test.py
Original file line number Diff line number Diff line change
Expand Up @@ -14,19 +14,19 @@
class CORInspectionDialogModelTest(unittest.TestCase):
def test_construct(self):
images = generate_images()
m = CORInspectionDialogModel(images, 5, ScalarCoR(20), ReconstructionParameters('FBP_CUDA', 'ram-lak'))
m = CORInspectionDialogModel(images, 5, ScalarCoR(20), ReconstructionParameters('FBP_CUDA', 'ram-lak'), False)
npt.assert_equal(m.sino, images.sino(5))
self.assertEqual(m.cor_extents, (0, 9))
self.assertEqual(m.proj_angles.value.shape, (10, ))

def test_start_cor_step(self):
images = generate_images()
m = CORInspectionDialogModel(images, 5, ScalarCoR(20), ReconstructionParameters('FBP_CUDA', 'ram-lak'))
m = CORInspectionDialogModel(images, 5, ScalarCoR(20), ReconstructionParameters('FBP_CUDA', 'ram-lak'), False)
self.assertEqual(images.width * 0.05, m.step)

def test_current_cor(self):
images = generate_images()
m = CORInspectionDialogModel(images, 5, ScalarCoR(20), ReconstructionParameters('FBP_CUDA', 'ram-lak'))
m = CORInspectionDialogModel(images, 5, ScalarCoR(20), ReconstructionParameters('FBP_CUDA', 'ram-lak'), False)
m.centre_value = 5
m.step = 1
self.assertEqual(m.cor(ImageType.LESS), 4)
Expand All @@ -35,7 +35,7 @@ def test_current_cor(self):

def test_adjust_cor(self):
images = generate_images()
m = CORInspectionDialogModel(images, 5, ScalarCoR(20), ReconstructionParameters('FBP_CUDA', 'ram-lak'))
m = CORInspectionDialogModel(images, 5, ScalarCoR(20), ReconstructionParameters('FBP_CUDA', 'ram-lak'), False)
m.centre_value = 5
m.step = 1

Expand Down

0 comments on commit 89232d0

Please sign in to comment.