Skip to content
3 changes: 2 additions & 1 deletion smarttub/api.py
Original file line number Diff line number Diff line change
Expand Up @@ -465,6 +465,7 @@ def __init__(self, spa: Spa, **properties):

self.intensity = properties["intensity"]
self.mode = self.LightMode[properties["mode"]]
self.cycleSpeed = properties["cycleSpeed"]
self.properties = properties

async def set_mode(self, mode: LightMode, intensity: int):
Expand All @@ -480,7 +481,7 @@ async def turn_off(self):
await self.set_mode(self.LightMode.OFF, 0)

def __str__(self):
return f"<SpaLight {self.zone}: {self.mode.name} (R {self.red}/G {self.green}/B {self.blue}/W {self.white}) @ {self.intensity}>"
return f"<SpaLight {self.zone}: {self.mode.name} {self.cycleSpeed} (R {self.red}/G {self.green}/B {self.blue}/W {self.white}) @ {self.intensity}>"


class SpaReminder:
Expand Down
3 changes: 3 additions & 0 deletions tests/test_light.py
Original file line number Diff line number Diff line change
Expand Up @@ -14,6 +14,9 @@ def lights(mock_spa):
"color": {"blue": 0, "green": 0, "red": 0, "white": 0},
"intensity": 0 if mode == SpaLight.LightMode.OFF else 50,
"mode": mode.name,
"cycleSpeed": 0,
"exterior": False,
"irt": None,
"zone": i + 1,
},
)
Expand Down