Add tests for str formatting of Price and Power types

Signed-off-by: Sahas Subramanian <[email protected]>

Author: shsms

tests/test_types.py

@@ -315,6 +315,18 @@ def test_price_from_pb() -> None:
     )
 
 
+def test_price_formatting() -> None:
+    """Test the client price type formatting."""
+    price = Price(amount=Decimal("800"), currency=Currency.JPY)
+    assert str(price) == "800 JPY"
+    price = Price(amount=Decimal("-100.3"), currency=Currency.CHF)
+    assert str(price) == "-100.3 CHF"
+    price = Price(amount=Decimal("-103.00"), currency=Currency.USD)
+    assert str(price) == "-103.00 USD"
+    price = Price(amount=Decimal("400.123"), currency=Currency.EUR)
+    assert str(price) == "400.123 EUR"
+
+
 def test_power_to_pb() -> None:
     """Test the client power type conversions to protobuf."""
     assert_conversion_to_pb(
@@ -333,6 +345,18 @@ def test_power_from_pb() -> None:
     )
 
 
+def test_power_formatting() -> None:
+    """Test the client power type formatting."""
+    power = Power(mw=Decimal("5"))
+    assert str(power) == "5 MW"
+    power = Power(mw=Decimal("5.1"))
+    assert str(power) == "5.1 MW"
+    power = Power(mw=Decimal("5.100"))
+    assert str(power) == "5.100 MW"
+    power = Power(mw=Decimal("-5.123"))
+    assert str(power) == "-5.123 MW"
+
+
 def test_delivery_duration_to_pb() -> None:
     """Test the client delivery duration type conversions to protobuf."""
     assert_conversion_to_pb(