Skip to content

Commit

Permalink
Fixup Recon vol name tests
Browse files Browse the repository at this point in the history
  • Loading branch information
JackEAllen committed May 30, 2024
1 parent 7563bb2 commit b4f2a57
Showing 1 changed file with 6 additions and 2 deletions.
8 changes: 6 additions & 2 deletions mantidimaging/gui/windows/recon/test/presenter_test.py
Original file line number Diff line number Diff line change
Expand Up @@ -346,7 +346,7 @@ def test_on_stack_reconstruct_slice_done(self):
test_data = ImageStack(np.ndarray(shape=(200, 250), dtype=np.float32))
test_data.record_operation = mock.Mock()
task_mock = mock.Mock(result=test_data, error=None)
recon_params = ReconstructionParameters("FBP_CUDA", "ram-lak", 10)
recon_params = ReconstructionParameters("gridrec", "ram-lak", 10)
self.view.recon_params.return_value = recon_params
self.presenter._get_slice_index = mock.Mock(return_value=7)
self.presenter._on_stack_reconstruct_slice_done(task_mock)
Expand Down Expand Up @@ -417,6 +417,8 @@ def test_infs_removed_from_recon_volume(self):
task = mock.Mock()
task.error = None
task.result.data = np.array([np.inf, -np.inf])
recon_params = ReconstructionParameters("gridrec", "ram-lak", 10)
self.view.recon_params.return_value = recon_params

self.presenter._on_volume_recon_done(task)
assert not np.isinf(self.view.show_recon_volume.call_args[0][0].data).any()
Expand All @@ -425,7 +427,7 @@ def test_infs_removed_from_recon_slice(self):
task = mock.Mock()
task.error = None
task.result.data = np.array([np.inf, -np.inf])
recon_params = ReconstructionParameters("FBP_CUDA", "ram-lak", 10)
recon_params = ReconstructionParameters("gridrec", "ram-lak", 10)
self.view.recon_params.return_value = recon_params

self.presenter._on_stack_reconstruct_slice_done(task)
Expand All @@ -436,6 +438,8 @@ def test_on_volume_recon_done_enables_buttons(self, _, error):
task = mock.Mock()
task.error = error
task.result.data = np.ones((5, 10, 10))
recon_params = ReconstructionParameters("gridrec", "ram-lak", 10)
self.view.recon_params.return_value = recon_params

self.presenter._on_volume_recon_done(task)
self.view.set_recon_buttons_enabled.assert_called_once_with(True)
Expand Down

0 comments on commit b4f2a57

Please sign in to comment.