def testCalculateFeaturesReturnsCorrectlySizedArray(self): pongGame = Pong() numberOfFeatures = 3 player = 1 action = NOTHING result = len(pongGame.getFeatures(player, action)) self.assertEqual(numberOfFeatures, result)
def testCalculateFeaturesReturnsNotNoneObjects(self): pongGame = Pong() player = 1 action = NOTHING result = pongGame.getFeatures(player, action) for i in result: self.assertIsNotNone(i)