def test_plot_save_saves_plot(mock_save, dummy_data): sph_obj, _, m, _, el, ss = dummy_data cl = StaticGeodesicPlotter(m) cl.plot(sph_obj, el, ss) name = "test_plot.png" cl.save(name) mock_save.assert_called_with(name)
def test_plot_save_saves_plot(mock_save, dummy_data): geodesic = dummy_data cl = StaticGeodesicPlotter() cl.plot(geodesic) name = "test_plot.png" cl.save(name) mock_save.assert_called_with(name)
def test_plot_geod_save_saves_parametric_plot(mock_save, dummy_geod): geod = dummy_geod sgpl = StaticGeodesicPlotter() sgpl.parametric_plot(geod) name = "Parametric.png" sgpl.save() mock_save.assert_called_with(name) assert sgpl.ax.name != "3d"
def test_plot_save_saves_plot(mock_save): r = [306 * u.m, np.pi / 2 * u.rad, np.pi / 2 * u.rad] v = [0 * u.m / u.s, 0 * u.rad / u.s, 951.0 * u.rad / u.s] m = 4e24 * u.kg el = 0.002 ss = 0.5e-6 cl = StaticGeodesicPlotter(m) cl.plot(r, v, el, ss) name = "test_plot.png" cl.save(name) mock_save.assert_called_with(name)