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)