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()
