umath: Use umath for math.

Do this for all Powered Up examples.

examples/pup/hub_cityhub/light_animate.py

@@ -1,7 +1,7 @@
 from pybricks.hubs import CityHub
 from pybricks.parameters import Color
 from pybricks.tools import wait
-from math import sin, pi
+from umath import sin, pi
 
 # Initialize the hub.
 hub = CityHub()

examples/pup/hub_primehub/light_animate.py

@@ -1,7 +1,7 @@
 from pybricks.hubs import PrimeHub
 from pybricks.parameters import Color
 from pybricks.tools import wait
-from math import sin, pi
+from umath import sin, pi
 
 # Initialize the hub.
 hub = PrimeHub()

examples/pup/hub_shared/light_animate.py

@@ -2,7 +2,7 @@
 from pybricks.hubs import ExampleHub
 from pybricks.parameters import Color
 from pybricks.tools import wait
-from math import sin, pi
+from umath import sin, pi
 
 # Initialize the hub.
 hub = ExampleHub()

examples/pup/hub_technichub/light_animate.py

@@ -1,7 +1,7 @@
 from pybricks.hubs import TechnicHub
 from pybricks.parameters import Color
 from pybricks.tools import wait
-from math import sin, pi
+from umath import sin, pi
 
 # Initialize the hub.
 hub = TechnicHub()

examples/pup/light/math.py

@@ -2,9 +2,7 @@ from pybricks.pupdevices import Light
 from pybricks.parameters import Port
 from pybricks.tools import wait, StopWatch
 
-# The math module is part of standard MicroPython:
-# https://docs.micropython.org/en/latest/library/math.html
-from math import pi, cos
+from umath import pi, cos
 
 # Initialize the light and a StopWatch.
 light = Light(Port.A)

examples/pup/sensor_ultrasonic/math.py

@@ -2,9 +2,7 @@ from pybricks.pupdevices import UltrasonicSensor
 from pybricks.parameters import Port
 from pybricks.tools import wait, StopWatch
 
-# The math module is part of standard MicroPython:
-# https://docs.micropython.org/en/latest/library/math.html
-from math import pi, sin
+from umath import pi, sin
 
 # Initialize the sensor.
 eyes = UltrasonicSensor(Port.A)