def test_simple_plot_manual_planes_two_files(tmp_path):
figs = plot(
files=[
str(INPUT / "beta_phase_{0}.tfs"),
str(INPUT / "beta_amplitude_{0}.tfs")
],
same_axes=["files"],
x_columns=["S"],
y_columns=["BET{0}"],
error_columns=None,
planes=["X", "Y"],
x_labels=["Location [m]"],
file_labels=[r"$\beta$ from phase", r"$\beta$ from amplitude"],
# column_labels=[r'$\beta_{0}$'], # would have correct axes-labels but also bx in legend
column_labels=[""], # removes BETX BETY from legend-names
y_labels=[
[r"$\beta_x$"],
[r"$\beta_y$"],
], # manual axes labels (outer = figures, inner = axes)
output=tmp_path,
show=False,
single_legend=True,
change_marker=True,
)
assert len(figs) == 2
for fig in figs.values():
assert len(fig.axes) == 1
def simple_plot_tfs(**kwargs):
default_args = dict(
files=[INPUT / "orbit_{0}.tfs"],
x_columns=["S"],
y_columns=["{0}"],
error_columns=["ERR{0}"],
planes=["X", "Y"],
show=False,
# same_axes='planes',
# same_figure='planes',
single_legend=True,
change_marker=True,
)
default_args.update(kwargs)
return plot(**default_args)