def _setLabels1D(axes, workspace, indices=None):
'''
helper function to automatically set axes labels for 1D plots
'''
labels = get_axes_labels(workspace, indices)
axes.set_xlabel(labels[1])
axes.set_ylabel(labels[0])
def _set_labels_3d(axes, workspace, indices=None):
"""
Helper function to automatically set axis labels for 3D plots
"""
labels = get_axes_labels(workspace, indices)
axes.set_xlabel(labels[1])
axes.set_ylabel(labels[2])
axes.set_zlabel(labels[0])
def _setLabels2D(axes, workspace, indices=None, transpose=False):
'''
helper function to automatically set axes labels for 2D plots
'''
labels = get_axes_labels(workspace, indices)
if transpose:
axes.set_xlabel(labels[2])
axes.set_ylabel(labels[1])
else:
axes.set_xlabel(labels[1])
axes.set_ylabel(labels[2])
axes.set_title(labels[-1])
def test_get_axes_label_2d_MDWS(self):
axs = funcs.get_axes_labels(self.ws_MD_2d)
# should get the first two dimension labels only
self.assertEqual(axs, ('Intensity', 'Dim1 ($\\AA^{-1}$)', 'Dim2 (EnergyTransfer)'))
def test_get_axes_labels(self):
axs = funcs.get_axes_labels(self.ws2d_histo)
self.assertEqual(axs, ('', 'Wavelength ($\\AA$)', 'Energy transfer ($meV$)'))
def test_get_axes_label_2d_MDWS_indices(self):
axs = funcs.get_axes_labels(self.ws_MD_2d, indices=(0,slice(None),0))
# should get the first two dimension labels only
self.assertEqual(axs, ('Intensity', 'Dim2 (EnergyTransfer)', 'Dim1=-2.4; Dim3=0.0;'))
def test_get_axes_label_2d_MDWS(self):
axs = funcs.get_axes_labels(self.ws_MD_2d)
# should get the first two dimension labels only
self.assertEqual(
axs,
('Intensity', 'Dim1 ($\\AA^{-1}$)', 'Dim2 (EnergyTransfer)', ''))
def test_y_units_for_non_distribution_and_autodist_on_with_latex(self):
ws = self.ws2d_non_distribution
labels = funcs.get_axes_labels(ws, normalize_by_bin_width=True)
self.assertEqual(labels[0], 'Counts (microAmp.hour $\\AA$)$^{-1}$')
def test_y_units_for_non_distribution_and_autodist_on_with_ascii(self):
ws = self.ws2d_non_distribution
labels = funcs.get_axes_labels(ws,
normalize_by_bin_width=True,
use_latex=False)
self.assertEqual(labels[0], 'Counts per microAmp.hour per Angstrom')
def test_y_units_correct_on_distribution_workspace(self):
ws = self.ws2d_distribution
labels = funcs.get_axes_labels(ws)
self.assertEqual(labels[0], 'Counts (microAmp.hour)$^{-1}$')
def test_get_axes_labels(self):
axs = funcs.get_axes_labels(self.ws2d_histo)
self.assertEqual(axs, ('($\\AA$)$^{-1}$', 'Wavelength ($\\AA$)',
'Energy transfer ($meV$)'))