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()