def test_to_array_spectrum(axes): s = LumiSpectrum(arange(5)) a = s.to_array(axes=axes) if axes: assert_array_equal(s.axes_manager[0].axis, a[:, 0]) else: assert_array_equal(s.data, a)
def test_to_array_linescan(axes, transpose): s = LumiSpectrum(arange(20).reshape((4, 5))) a = s.to_array(axes=axes, transpose=transpose) if axes: if transpose: assert_array_equal(s.axes_manager[0].axis, a[0, 1:]) assert_array_equal(s.axes_manager[1].axis, a[1:, 0]) assert_array_equal(s.data.T, a[1:, 1:]) else: assert_array_equal(s.axes_manager[1].axis, a[0, 1:]) assert_array_equal(s.axes_manager[0].axis, a[1:, 0]) assert_array_equal(s.data, a[1:, 1:]) else: if transpose: assert_array_equal(s.data.T, a) else: assert_array_equal(s.data, a)
def test_to_array_dimension_error(): s = LumiSpectrum(arange(60).reshape((3, 4, 5))) with raises(NotImplementedError): s.to_array()