def test_basic_orbit_and_trajectory_plotting(): fig, ax = plt.subplots() plotter = StaticOrbitPlotter(ax=ax) plotter.plot(iss) plotter.plot_trajectory(molniya.sample(), label="Molniya") return fig
def test_basic_trajectory_plotting(): fig, ax = plt.subplots() plotter = StaticOrbitPlotter(ax=ax) plotter.set_attractor(Earth) plotter.set_orbit_frame(iss) plotter.plot_trajectory(iss.sample()) return fig
def test_plot_trajectory_sets_label(): op = StaticOrbitPlotter() earth = Orbit.from_body_ephem(Earth) mars = Orbit.from_body_ephem(Mars) trajectory = earth.sample() op.plot(mars, label="Mars") op.plot_trajectory(trajectory, label="Earth") legend = plt.gca().get_legend() assert legend.get_texts()[1].get_text() == "Earth"
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
def test_redraw_keeps_trajectories(): # See https://github.com/poliastro/poliastro/issues/518 op = StaticOrbitPlotter() earth = Orbit.from_body_ephem(Earth) mars = Orbit.from_body_ephem(Mars) trajectory = earth.sample() op.plot(mars, label="Mars") op.plot_trajectory(trajectory, label="Earth") assert len(op.trajectories) == 2 op.set_frame(*mars.pqw()) assert len(op.trajectories) == 2