def test_period_graph_update(self, beeman, plot, show):
try:
system = SolarSystem(3600, 10.175 * 10**3, Options.NORMAL_RUN,
"CelestialObjects")
Animation.periods_graph(100, system)
except ZeroDivisionError as err:
self.assertEqual(100, beeman.call_count)
def test_period_graph_plot(self, plot, show):
system = SolarSystem(1000000, 10.175 * 10**3, Options.NORMAL_RUN,
"CelestialObjects")
system.celestial_bodies = system.celestial_bodies[0:2]
Animation.periods_graph(10, system)
show.assert_called_once()