示例#1
0
    def test_sliceviewer_MDH(self):
        presenter = SliceViewer(None, model=self.model, view=self.view)

        # setup calls
        self.assertEqual(self.model.get_ws.call_count, 1)
        self.assertEqual(self.model.get_properties.call_count, 1)
        self.assertEqual(
            self.view.data_view.dimensions.get_slicepoint.call_count,
            2)  # extra call during init of pres
        self.assertEqual(self.view.data_view.plot_MDH.call_count, 1)

        # new_plot
        self.model.reset_mock()
        self.view.reset_mock()
        presenter.new_plot()
        self.assertEqual(self.model.get_ws.call_count, 1)
        self.assertEqual(
            self.view.data_view.dimensions.get_slicepoint.call_count, 1)
        self.assertEqual(self.view.data_view.plot_MDH.call_count, 1)

        # update_plot_data
        self.model.reset_mock()
        self.view.reset_mock()
        presenter.update_plot_data()
        self.assertEqual(self.model.get_data.call_count, 1)
        self.assertEqual(
            self.view.data_view.dimensions.get_slicepoint.call_count, 1)
        self.assertEqual(self.view.data_view.update_plot_data.call_count, 1)
示例#2
0
    def runTest(self):
        HelperTestingClass.__init__(self)
        hkl_ws = create_hkl_ws()
        pres = SliceViewer(hkl_ws)

        # not transpose
        pres.view.data_view.dimensions.transpose = False
        pres.update_plot_data()
        extent = pres.view.data_view.image.get_extent()
        assert_allclose(extent, [-10.0, 10.0, -9.0, 9.0])

        # transpose
        pres.view.data_view.dimensions.transpose = True
        pres.update_plot_data()
        extent = pres.view.data_view.image.get_extent()

        # Should be the same as before as transposition is handled in the model
        assert_allclose(extent, [-10.0, 10.0, -9.0, 9.0])
        assert_allclose(extent[0:2], list(pres.view.data_view.ax.get_xlim()))
        assert_allclose(extent[2:], list(pres.view.data_view.ax.get_ylim()))

        pres.view.close()
示例#3
0
    def test_sliceviewer_MDE(self):
        self.patched_deps[
            "WorkspaceInfo"].get_ws_type.return_value = WS_TYPE.MDE

        presenter = SliceViewer(None, model=self.model, view=self.view)

        # setup calls
        self.assertEqual(self.model.get_ws_MDE.call_count, 1)
        self.assertEqual(self.model.get_properties.call_count, 1)
        self.assertEqual(
            self.view.data_view.dimensions.get_slicepoint.call_count,
            2)  # extra call during init of pres
        self.assertEqual(
            self.view.data_view.dimensions.get_bin_params.call_count, 1)
        self.assertEqual(self.view.data_view.plot_MDH.call_count, 1)

        # new_plot
        self.model.reset_mock()
        self.view.reset_mock()
        presenter.new_plot()
        self.assertEqual(self.model.get_ws_MDE.call_count, 1)
        self.assertEqual(
            self.view.data_view.dimensions.get_slicepoint.call_count, 1)
        self.assertEqual(
            self.view.data_view.dimensions.get_bin_params.call_count, 1)
        self.assertEqual(self.view.data_view.plot_MDH.call_count, 1)

        # update_plot_data
        self.model.reset_mock()
        self.view.reset_mock()
        presenter.update_plot_data()
        self.assertEqual(self.model.get_data.call_count, 1)
        self.assertEqual(
            self.view.data_view.dimensions.get_slicepoint.call_count, 1)
        self.assertEqual(
            self.view.data_view.dimensions.get_bin_params.call_count, 1)
        self.assertEqual(self.view.data_view.update_plot_data.call_count, 1)