def test_animate_calls_figure(mock_figure):
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 = ScatterGeodesicPlotter(m)
cl.animate(r, v, el, ss)
mock_figure.assert_called_with()
def test_animate_calls_figure(mock_figure, dummy_data):
sph_obj, _, m, _, el, ss = dummy_data
cl = ScatterGeodesicPlotter(m)
cl.animate(sph_obj, el, ss)
mock_figure.assert_called_with()