Use FloatInt in PowerTransformer

Signed-off-by: Leandro Lucarella <luca-frequenz@llucax.com>

Author: llucax

src/frequenz/client/common/microgrid/electrical_components/_power_transformer.py

@@ -5,6 +5,7 @@
 
 import dataclasses
 
+from ..._float import FloatInt
 from ._electrical_component import ElectricalComponent
 
 
@@ -22,13 +23,13 @@ class PowerTransformer(ElectricalComponent):
         than the input power.
     """
 
-    primary_voltage: float
+    primary_voltage: FloatInt
     """The primary voltage of the transformer, in volts.
 
     This is the input voltage that is stepped up or down.
     """
 
-    secondary_voltage: float
+    secondary_voltage: FloatInt
     """The secondary voltage of the transformer, in volts.
 
     This is the output voltage that is the result of stepping the primary

tests/microgrid/electrical_components/test_power_transformer.py

@@ -5,6 +5,7 @@
 
 import pytest
 
+from frequenz.client.common import FloatInt
 from frequenz.client.common.microgrid import MicrogridId
 from frequenz.client.common.microgrid.electrical_components import (
     ElectricalComponentId,
@@ -25,13 +26,14 @@ def microgrid_id() -> MicrogridId:
 
 
 @pytest.mark.parametrize(
-    "primary, secondary", [(400.0, 230.0), (0.0, 0.0), (230.0, 400.0), (-230.0, -400.0)]
+    "primary, secondary",
+    [(400.0, 230.0), (0.0, 0.0), (230.0, 400.0), (-230.0, -400.0), (400, 230)],
 )
 def test_creation_ok(
     component_id: ElectricalComponentId,
     microgrid_id: MicrogridId,
-    primary: float,
-    secondary: float,
+    primary: FloatInt,
+    secondary: FloatInt,
 ) -> None:
     """Test PowerTransformer component initialization with different voltages."""
     power_transformer = PowerTransformer(