fix layout test

reactive-python · Sep 4, 2020 · e9b83ea · e9b83ea
1 parent a703d5f
commit e9b83ea
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/tests/test_core/test_layout.py b/tests/test_core/test_layout.py
@@ -234,11 +234,11 @@ def AnElement():
 
         await layout.render()
         try:
-            asyncio.wait_for(
-                [layout.render()],
+            await asyncio.wait_for(
+                layout.render(),
                 timeout=0.1,  # this should have been plenty of time
             )
-        except asyncio.CancelledError:
+        except asyncio.TimeoutError:
             pass  # the render should still be rendering since we only update once
 
         assert run_count.current == 2