typing

Author: James Souter

tests/test_attribute.py

@@ -144,6 +144,8 @@ async def get(self, uri: str):
         elif uri == "status/float_parameter":
             value = self._float_value
             self._float_value += 1
+        else:
+            raise RuntimeError()
         return value
 
     async def set(self, uri: str, value: float | int):
@@ -174,15 +176,16 @@ async def update(
             attr: AttrR[NumberT, DemoParameterAttributeIORef],
         ):
             value = await attr.io_ref.connection.get(attr.io_ref.uri)
-            await attr.set(value)
+            await attr.set(value)  # type: ignore
 
         async def send(
             self,
             attr: AttrW[NumberT, DemoParameterAttributeIORef],
             value: NumberT,
         ) -> None:
             await attr.io_ref.connection.set(attr.io_ref.uri, value)
-            await self.update(attr)
+            if isinstance(attr, AttrRW):
+                await self.update(attr)
 
     class DemoParameterController(Controller):
         ro_int_parameter: AttrR
@@ -196,6 +199,7 @@ async def initialise(self):
             example_introspection_response = await self._connection.get(
                 "config/introspect_api"
             )
+            assert isinstance(example_introspection_response, list)
             for parameter_response in example_introspection_response:
                 try:
                     ro = parameter_response["read_only"]
@@ -250,7 +254,9 @@ class MyController(Controller):
 
     class SimpleAttributeIO(AttributeIO[T, AttributeIORef]):
         async def update(self, attr):
-            await attr.set(100)
+            match attr:
+                case AttrR(datatype=Int()):
+                    await attr.set(100)
 
     with pytest.raises(
         RuntimeError, match="More than one AttributeIO class handles AttributeIORef"