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)