fix flaky test

Author: rmorshea

src/py/reactpy/reactpy/testing/common.py

@@ -67,7 +67,7 @@ async def until(
                 break
             elif (time.time() - started_at) > timeout:  # nocov
                 msg = f"Expected {description} after {timeout} seconds - last value was {result!r}"
-                raise TimeoutError(msg)
+                raise asyncio.TimeoutError(msg)
 
     async def until_is(
         self,

src/py/reactpy/tests/test_core/test_hooks.py

@@ -1037,9 +1037,11 @@ def SetStateDuringRender():
         # we expect a second render to be triggered in the background
         await poll(lambda: render_count.current).until_equals(2)
 
-        # there should be no more renders that happen
-        with pytest.raises(asyncio.TimeoutError):
-            await poll(lambda: render_count.current).until_equals(3, timeout=0.1)
+        # give an opportunity for a render to happen if it were to.
+        await asyncio.sleep(0.1)
+
+    # however, we don't expect any more renders
+    assert render_count.current == 2
 
 
 @pytest.mark.skipif(not REACTPY_DEBUG_MODE.current, reason="only logs in debug mode")