def test_scatterplot_show(self, mock):
"""Test used to prove that plt.show() is called once in scatter_plot method
uses mocking
"""
self.system_probe = SolarSystem(3600, 10000, Options.PROBE_RUN,
"CelestialObjects")
animate = Animation(self.system_probe)
animate.scatter_plot(100)
mock.assert_called_once()
def test_scatterplot_update(self, mock, mock2):
"""Test used to prove that update_beeman() is called 100 times in scatter_plot method
if the parameter passed to scatter_plot is 100.
uses mocking
"""
self.system_probe = SolarSystem(3600, 10000, Options.PROBE_RUN,
"CelestialObjects")
animate = Animation(self.system_probe)
animate.scatter_plot(100)
mock.assert_called()
self.assertEqual(mock.call_count, 100)