def setUp(self):
self.sv_view = mock.create_autospec(StackVisualiserView)
self.sv_view.current_roi = self.ROI_PARAMETER
self.sv_presenter = StackVisualiserPresenter(self.sv_view, self.test_data)
self.sv_view.presenter = self.sv_presenter
self.model = FiltersWindowModel(mock.MagicMock())
def setUp(self):
self.model = ReconstructWindowModel(CorTiltPointQtModel())
# Mock stack
self.stack = mock.create_autospec(StackVisualiserView)
data = Images(data=np.ndarray(shape=(10, 128, 256), dtype=np.float32))
self.stack.presenter = StackVisualiserPresenter(self.stack, data)
self.model.initial_select_data(self.stack)
class StackVisualiserPresenterTest(unittest.TestCase):
test_data: Images
def __init__(self, *args, **kwargs):
super(StackVisualiserPresenterTest, self).__init__(*args, **kwargs)
def setUp(self):
self.test_data = th.generate_images()
# mock the view so it has the same methods
self.view = mock.create_autospec(StackVisualiserView)
self.presenter = StackVisualiserPresenter(self.view, self.test_data)
def tearDown(self) -> None:
try:
self.test_data.free_memory()
except FileNotFoundError:
# the test has deleted the data manually
pass
@classmethod
def setUpClass(cls) -> None:
for a in sa.list():
sa.delete(a.name.decode("utf-8"))
@classmethod
def tearDownClass(cls) -> None:
assert len(
sa.list()
) == 0, f"Not all shared arrays have been freed. Leftover: {sa.list()}"
def test_get_image(self):
index = 3
test_data = self.test_data
img = self.presenter.get_image(index)
npt.assert_equal(test_data.data[index], img.data[0])
def test_delete_data(self):
self.presenter.images = th.generate_images(automatic_free=False)
self.presenter.delete_data()
self.assertIsNone(self.presenter.images, None)
def test_notify_refresh_image_normal_image_mode(self):
self.presenter.image_mode = SVImageMode.NORMAL
self.presenter.notify(SVNotification.REFRESH_IMAGE)
self.assertIs(self.view.image, self.presenter.images.data,
"Image should have been set as sample images")
def test_notify_refresh_image_averaged_image_mode(self):
self.presenter.image_mode = SVImageMode.SUMMED
self.presenter.notify(SVNotification.REFRESH_IMAGE)
self.assertIs(self.view.image, self.presenter.summed_image,
"Image should have been set as averaged image")
def setUp(self):
# Mock view
self.make_view()
self.presenter = ReconstructWindowPresenter(self.view, None)
# Mock stack
self.sv_view = mock.create_autospec(StackVisualiserView)
data = Images(data=np.ndarray(shape=(128, 10, 128), dtype=np.float32))
self.sv_view.presenter = StackVisualiserPresenter(self.sv_view, data)
self.presenter.model.initial_select_data(self.sv_view)
self.view.get_stack_visualiser = mock.Mock(return_value=self.sv_view)
import uuid
self.uuid = uuid.uuid4()
def setUp(self):
self.test_data = th.generate_images()
# mock the view so it has the same methods
self.view = mock.create_autospec(StackVisualiserView)
self.presenter = StackVisualiserPresenter(self.view, self.test_data)
class StackVisualiserPresenterTest(unittest.TestCase):
test_data: Images
def setUp(self):
self.test_data = th.generate_images()
# mock the view so it has the same methods
self.view = mock.create_autospec(StackVisualiserView)
self.presenter = StackVisualiserPresenter(self.view, self.test_data)
self.presenter.model = mock.Mock()
def test_get_image(self):
index = 3
test_data = self.test_data
img = self.presenter.get_image(index)
npt.assert_equal(test_data.data[index], img.data[0])
def test_delete_data(self):
self.presenter.images = th.generate_images()
self.presenter.delete_data()
self.assertIsNone(self.presenter.images, None)
def test_notify_refresh_image_normal_image_mode(self):
self.presenter.image_mode = SVImageMode.NORMAL
self.presenter.notify(SVNotification.REFRESH_IMAGE)
self.assertIs(self.view.image, self.presenter.images.data,
"Image should have been set as sample images")
def test_notify_refresh_image_averaged_image_mode(self):
self.presenter.image_mode = SVImageMode.SUMMED
self.presenter.notify(SVNotification.REFRESH_IMAGE)
self.assertIs(self.view.image, self.presenter.summed_image,
"Image should have been set as averaged image")
def test_notify_toggle_image_mode_normal_to_summed(self):
self.presenter.image_mode = SVImageMode.SUMMED
self.presenter.notify(SVNotification.TOGGLE_IMAGE_MODE)
assert self.presenter.image_mode is SVImageMode.NORMAL
self.presenter.model.sum_images.assert_not_called()
def test_notify_toggle_image_mode_summed_to_normal(self):
self.presenter.image_mode = SVImageMode.NORMAL
self.presenter.notify(SVNotification.TOGGLE_IMAGE_MODE)
assert self.presenter.image_mode is SVImageMode.SUMMED
def test_notify_toggle_image_mode_sets_summed_image(self):
self.presenter.image_mode = SVImageMode.NORMAL
self.summed_image = None
self.presenter.notify(SVNotification.TOGGLE_IMAGE_MODE)
assert self.presenter.summed_image == self.presenter.model.sum_images.return_value
self.presenter.model.sum_images.assert_called_once_with(
self.presenter.images.data)
def test_notify_toggle_image_mode_does_not_set_summed_image(self):
self.presenter.image_mode = SVImageMode.NORMAL
self.presenter.summed_image = self.presenter.images.data[0]
self.presenter.notify(SVNotification.TOGGLE_IMAGE_MODE)
self.presenter.model.sum_images.assert_not_called()
def test_get_num_images(self):
assert self.presenter.get_num_images(
) == self.presenter.images.num_projections
def test_find_image_from_angle_returns_matching_index(self):
angle_rad = self.presenter.images.projection_angles().value[1]
angle_deg = np.rad2deg(angle_rad)
index = self.presenter.find_image_from_angle(angle_deg)
self.assertAlmostEqual(
self.presenter.images.projection_angles().value[index], angle_rad)
def test_find_image_from_angle_returns_next_index(self):
angle = (self.presenter.images.projection_angles().value[1] +
self.presenter.images.projection_angles().value[2]) * 0.5
angle = np.rad2deg(angle)
index = self.presenter.find_image_from_angle(angle)
assert index == 2
def test_find_image_from_angle_returns_number_of_values(self):
angle = self.presenter.images.projection_angles().value[-1] * 2
angle = np.rad2deg(angle)
assert self.presenter.find_image_from_angle(angle) == len(
self.presenter.images.projection_angles().value)
@patch("mantidimaging.gui.windows.stack_visualiser.presenter.getLogger")
def test_notify_exception_log(self, get_logger_mock):
self.presenter.refresh_image = mock.Mock()
self.presenter.refresh_image.side_effect = Exception
self.presenter.notify(SVNotification.REFRESH_IMAGE)
get_logger_mock.return_value.exception.assert_called_once_with(
"Notification handler failed")
def test_get_parameter_value_returns_roi(self):
assert self.presenter.get_parameter_value(
SVParameters.ROI) == self.presenter.view.current_roi
def test_get_parameter_value_raises_value_error(self):
with self.assertRaises(ValueError):
self.presenter.get_parameter_value(7)
