Apply suggestions

Author: davidbrochart

tests/test_comm_awareness.py

@@ -2,6 +2,7 @@
 
 import pytest
 from pycrdt import Awareness, Doc, YMessageType, create_awareness_message
+from ypywidgets.comm import CommWidget
 
 pytestmark = pytest.mark.anyio
 
@@ -12,34 +13,32 @@ async def test_comm_provider_applies_awareness_frame(synced_widgets, context):
         remote_awareness = Awareness(Doc())
         remote_awareness.set_local_state({"role": "remote"})
         payload = remote_awareness.encode_awareness_update([remote_awareness.client_id])
-        frame = create_awareness_message(payload)
+        message = create_awareness_message(payload)
 
-        assert frame[0] == YMessageType.AWARENESS
+        assert message[0] == YMessageType.AWARENESS
 
-        local_widget._comm_provider._receive({"buffers": [frame]})
+        local_widget._comm_provider._receive({"buffers": [message]})
 
         remote_state = local_widget.awareness.states.get(remote_awareness.client_id)
         assert remote_state is not None
         assert remote_state.get("role") == "remote"
 
 
-async def test_comm_widget_exposes_provider_awareness(synced_widgets, context):
-    async with context:
-        widget = await synced_widgets.get_local_widget()
-        assert widget.awareness is widget._comm_provider.awareness
+async def test_comm_widget_exposes_provider_awareness():
+    widget = CommWidget()
+    assert widget.awareness is widget._comm_provider.awareness
 
 
 async def test_comm_widget_awareness_observe_and_unobserve(synced_widgets, context):
-    async with context:
-        widget = await synced_widgets.get_local_widget()
+    widget = CommWidget()
 
-        events: list[str] = []
-        sub_id = widget.awareness.observe(lambda topic, _: events.append(topic))
+    events: list[str] = []
+    sub_id = widget.awareness.observe(lambda topic, _: events.append(topic))
 
-        widget.awareness.set_local_state({"ping": 1})
-        assert events
+    widget.awareness.set_local_state({"ping": 1})
+    assert events
 
-        widget.awareness.unobserve(sub_id)
-        events.clear()
-        widget.awareness.set_local_state({"ping": 2})
-        assert events == []
+    widget.awareness.unobserve(sub_id)
+    events.clear()
+    widget.awareness.set_local_state({"ping": 2})
+    assert events == []