def test_savetxt_spectrum(axes, tmp_path):
s = LumiSpectrum(arange(5))
fname = tmp_path / "test.txt"
s.savetxt(fname, axes=axes)
s2 = loadtxt(fname)
if axes:
assert_array_equal(s.axes_manager[0].axis, s2[:, 0])
else:
assert_array_equal(s.data, s2)
def test_savetxt_linescan(axes, transpose, tmp_path):
s = LumiSpectrum(arange(20).reshape((4, 5)))
fname = tmp_path / "test.txt"
s.savetxt(fname, axes=axes, transpose=transpose)
s2 = loadtxt(fname)
if axes:
if transpose:
assert_array_equal(s.axes_manager[0].axis, s2[0, 1:])
assert_array_equal(s.axes_manager[1].axis, s2[1:, 0])
assert_array_equal(s.data.T, s2[1:, 1:])
else:
assert_array_equal(s.axes_manager[1].axis, s2[0, 1:])
assert_array_equal(s.axes_manager[0].axis, s2[1:, 0])
assert_array_equal(s.data, s2[1:, 1:])
else:
if transpose:
assert_array_equal(s.data.T, s2)
else:
assert_array_equal(s.data, s2)
def test_savetxt_dimension_error(tmp_path):
s = LumiSpectrum(arange(60).reshape((3, 4, 5)))
fname = tmp_path / "test.txt"
with raises(NotImplementedError):
s.savetxt(fname)