class TestGameEngine(TestCase):
def setUp(self):
pygame.init()
self.engine = GameEngine()
def tearDown(self):
pygame.quit()
def test_update(self):
self.engine.update()
self.assertEqual(1, self.engine.total_frames)
def test_restart(self):
self.engine.update()
self.engine.game_over = True
self.engine.restart()
self.assertEqual(0, self.engine.total_frames)
self.assertEqual(0, self.engine.total_frames_played)
