def test_plot_calls_draw_attractor_Manualscale(dummy_data):
geodesic = dummy_data
cl = StaticGeodesicPlotter(attractor_radius_scale=1500)
cl.plot(geodesic)
assert cl._attractor_present
assert cl.attractor_radius_scale == 1500
assert cl.get_curr_plot_radius != -1
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_calls_plt_show(mock_show, dummy_data):
geodesic = dummy_data
cl = StaticGeodesicPlotter()
cl.plot(geodesic)
cl.show()
mock_show.assert_called_with()
assert cl._attractor_present
def test_plot_calls_draw_attractor_AutoScale(dummy_data):
geodesic = dummy_data
cl = StaticGeodesicPlotter()
cl.plot(geodesic)
assert cl._attractor_present
assert cl.get_curr_plot_radius != -1
def test_animate_creates_ani(dummy_data):
geodesic = dummy_data
cl = StaticGeodesicPlotter()
cl.animate(geodesic, interval=10)
assert cl._attractor_present
assert cl.ani
def test_staticgeodesicplotter_has_axes(dummy_data):
geodesic = dummy_data
cl = StaticGeodesicPlotter()
assert isinstance(cl.ax, mpl.axes.SubplotBase)
assert cl.time.value == 0.0
assert cl._attractor_present is False