コード例 #1
0
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")
コード例 #2
0
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)