def test_whenCalculatingMagnitude_thenMagnitudeIsCorrect(self): position = Position(X_1, Y_1) magnitude = position.magnitude() expected_magnitude = sqrt(X_1 * X_1 + Y_1 * Y_1) self.assertEqual(expected_magnitude, magnitude)
def move( self, element_to_move: WorldElementSubclass, delta_position: Position, energy: Energy, ): try: self._move(element_to_move, delta_position) energy.remove(delta_position.magnitude()**2) except NoMoreEnergy: self._undo_move(element_to_move, delta_position) raise NotEnoughEnergyToMove()