1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
|
import pytest
from matplotlib import pyplot as plt
from poliastro.plotting import OrbitPlotter2D, OrbitPlotter3D
from poliastro.plotting.misc import plot_solar_system
@pytest.mark.parametrize("outer, expected", [(True, 8), (False, 4)])
def test_plot_solar_system_has_expected_number_of_orbits(outer, expected):
assert len(plot_solar_system(outer).trajectories) == expected
@pytest.mark.parametrize(
"use_3d, plotter_class", [(True, OrbitPlotter3D), (False, OrbitPlotter2D)]
)
def test_plot_solar_system_uses_expected_orbitplotter(use_3d, plotter_class):
assert isinstance(
plot_solar_system(use_3d=use_3d, interactive=True), plotter_class
)
def test_plot_solar_system_raises_error_if_use_3d_and_non_interactive():
with pytest.raises(ValueError) as excinfo:
plot_solar_system(use_3d=True)
assert "The static plotter does not support 3D" in excinfo.exconly()
@pytest.mark.mpl_image_compare
def test_plot_inner_solar_system_static(earth_perihelion):
plot_solar_system(outer=False, epoch=earth_perihelion)
return plt.gcf()
@pytest.mark.mpl_image_compare
def test_plot_outer_solar_system_static(earth_perihelion):
plot_solar_system(outer=True, epoch=earth_perihelion)
return plt.gcf()
|
