Type as str | None

Author: bouwew

plugwise/__init__.py

@@ -84,10 +84,10 @@ def __init__(
         self.smile_model_id: str | None = None
         self.smile_name: str = NONE
         self.smile_type: str = NONE
-        self.smile_version: str = NONE
+        self.smile_version: str | None = None
         self.smile_zigbee_mac_address: str | None = None
 
-    async def connect(self) -> str:
+    async def connect(self) -> str | None:
         """Connect to Plugwise device and determine its name, type and version."""
         result = await self._request(DOMAIN_OBJECTS)
         # Work-around for Stretch fw 2.7.18

tests/test_init.py

@@ -346,7 +346,7 @@ async def connect(
         # Connect to the smile
         try:
             version = await smile.connect()
-            assert version != pw_constants.NONE
+            assert version is not None
             return server, smile, client
         except (
             pw_exceptions.ConnectionFailedError,
@@ -429,7 +429,7 @@ async def connect_legacy(
         # Connect to the smile
         try:
             version = await smile.connect()
-            assert version != pw_constants.NONE
+            assert version is not None
             return server, smile, client
         except (
             pw_exceptions.ConnectionFailedError,