def test_plot_3d_trajectory_plots_a_trajectory():
frame = OrbitPlotter3D()
assert len(frame.trajectories) == 0
trajectory = churi.sample()
frame.set_attractor(Sun)
frame.plot_trajectory(trajectory)
assert len(frame.trajectories) == 1
assert frame._attractor == Sun
def test_plot_trajectory_sets_label():
expected_label = "67P"
op = StaticOrbitPlotter()
trajectory = churi.sample()
op.plot_body_orbit(Mars, J2000_TDB, label="Mars")
op.plot_trajectory(trajectory, label=expected_label)
legend = plt.gca().get_legend()
assert legend.get_texts()[1].get_text() == expected_label
def test_redraw_keeps_trajectories():
# See https://github.com/poliastro/poliastro/issues/518
op = StaticOrbitPlotter()
trajectory = churi.sample()
op.plot_body_orbit(Mars, J2000_TDB, label="Mars")
op.plot_trajectory(trajectory, label="67P")
assert len(op.trajectories) == 2
op.set_body_frame(Mars)
assert len(op.trajectories) == 2
