def _get_data_for_plot(axes, workspace, kwargs, with_dy=False, with_dx=False): if isinstance(workspace, mantid.dataobjects.MDHistoWorkspace): (normalization, kwargs) = get_normalization(workspace, **kwargs) indices, kwargs = get_indices(workspace, **kwargs) (x, y, dy) = get_md_data1d(workspace, normalization, indices) dx = None axis = None else: axis = MantidAxType(kwargs.pop("axis", MantidAxType.SPECTRUM)) normalize_by_bin_width, kwargs = get_normalize_by_bin_width( workspace, axes, **kwargs) workspace_index, distribution, kwargs = get_wksp_index_dist_and_label( workspace, axis, **kwargs) if axis == MantidAxType.BIN: # Overwrite any user specified xlabel axes.set_xlabel("Spectrum") x, y, dy, dx = get_bins(workspace, workspace_index, with_dy) elif axis == MantidAxType.SPECTRUM: x, y, dy, dx = get_spectrum(workspace, workspace_index, normalize_by_bin_width, with_dy, with_dx) else: raise ValueError( "Axis {} is not a valid axis number.".format(axis)) indices = None return x, y, dy, dx, indices, axis, kwargs
def _get_data_for_plot(axes, workspace, kwargs, with_dy=False, with_dx=False): if isinstance(workspace, mantid.dataobjects.MDHistoWorkspace): (normalization, kwargs) = get_normalization(workspace, **kwargs) indices, kwargs = get_indices(workspace, **kwargs) (x, y, dy) = get_md_data1d(workspace, normalization, indices) dx = None else: axis = kwargs.pop("axis", MantidAxType.SPECTRUM) workspace_index, distribution, kwargs = get_wksp_index_dist_and_label(workspace, axis, **kwargs) if axis == MantidAxType.BIN: # Overwrite any user specified xlabel axes.set_xlabel("Spectrum") x, y, dy, dx = get_bins(workspace, workspace_index, with_dy) elif axis == MantidAxType.SPECTRUM: x, y, dy, dx = get_spectrum(workspace, workspace_index, distribution, with_dy, with_dx) else: raise ValueError("Axis {} is not a valid axis number.".format(axis)) indices = None return x, y, dy, dx, indices, kwargs
def test_get_bins_with_dy(self, ws): x, y, dy, dx = funcs.get_bins(ws, 1, withDy=True) self.assertTrue(np.array_equal([0, 1, 2, 3], x)) self.assertTrue(np.array_equal([2.0, 5.0, 8.0, 11.0], y)) self.assertTrue(np.array_equal([2.0, 5.0, 8.0, 11.0], dy))
def test_get_bins_with_dy(self, ws): x, y, dy, dx = funcs.get_bins(ws, 1, withDy=True) self.assertTrue(np.array_equal([0, 1, 2, 3], x)) self.assertTrue(np.array_equal([2.0, 5.0, 8.0, 11.0], y)) self.assertTrue(np.array_equal([2.0, 5.0, 8.0, 11.0], dy))