Add tests for thermal commands

Author: Steve-Tech

cros_ec_python/commands/thermal.py

@@ -3,11 +3,9 @@
 from ..cros_ec import CrOS_EC
 from ..constants.COMMON import *
 
-
 EC_CMD_THERMAL_SET_THRESHOLD: Final = 0x0050
 EC_CMD_THERMAL_GET_THRESHOLD: Final = 0x0051
 
-
 EC_CMD_THERMAL_AUTO_FAN_CTRL: Final = 0x0052
 
 
@@ -26,6 +24,4 @@ def thermal_auto_fan_ctrl(ec: CrOS_EC, fan_idx: UInt8 | None = None) -> None:
 
 EC_CMD_TMP006_GET_CALIBRATION: Final = 0x0053
 EC_CMD_TMP006_SET_CALIBRATION: Final = 0x0054
-
-
 EC_CMD_TMP006_GET_RAW: Final = 0x0055

tests/tests.py

@@ -1,6 +1,10 @@
 import unittest
 from general import *
 from features import *
+# These have some user input, uncomment if you're human.
+# from pwm import *
+# from leds import *
+from thermal import *
 
 
 if __name__ == '__main__':

tests/thermal.py

@@ -0,0 +1,16 @@
+import unittest
+from cros_ec_python import CrOS_EC, thermal as ec_thermal
+
+ec = CrOS_EC()
+
+
+class TestAutoFanControl(unittest.TestCase):
+    def test_version0(self):
+        ec_thermal.thermal_auto_fan_ctrl(ec)
+
+    def test_version1(self):
+        ec_thermal.thermal_auto_fan_ctrl(ec, 0)
+
+
+if __name__ == '__main__':
+    unittest.main()