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