diff --git a/src/airgradient/airgradient.py b/src/airgradient/airgradient.py index 0d2de17..cb11a52 100644 --- a/src/airgradient/airgradient.py +++ b/src/airgradient/airgradient.py @@ -107,6 +107,14 @@ async def set_led_bar_mode(self, led_bar_mode: LedBarMode) -> None: """Set LED bar mode on AirGradient device.""" await self._set_config("ledBarMode", led_bar_mode) + async def request_co2_calibration(self) -> None: + """Request CO2 calibration on AirGradient device.""" + await self._set_config("co2CalibrationRequested", value=True) + + async def request_led_bar_test(self) -> None: + """Request LED bar test on AirGradient device.""" + await self._set_config("ledBarTestRequested", value=True) + async def close(self) -> None: """Close open client session.""" if self.session and self._close_session: diff --git a/tests/test_airgradient.py b/tests/test_airgradient.py index 40bafef..4f78bda 100644 --- a/tests/test_airgradient.py +++ b/tests/test_airgradient.py @@ -126,6 +126,14 @@ async def test_config( lambda client: client.set_led_bar_mode(LedBarMode.CO2), {"ledBarMode": "co2"}, ), + ( + lambda client: client.request_co2_calibration(), + {"co2CalibrationRequested": True}, + ), + ( + lambda client: client.request_led_bar_test(), + {"ledBarTestRequested": True}, + ), ], ) async def test_setting_config(