specify TraitError in test (#85)

vispy · Nov 27, 2023 · a1b8b47 · a1b8b47
1 parent 3fd0dc6
commit a1b8b47
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/tests/test_widget.py b/tests/test_widget.py
@@ -10,6 +10,7 @@
 from pytest import raises
 from jupyter_rfb import RemoteFrameBuffer
 from jupyter_rfb._utils import Snapshot
+from traitlets import TraitError
 
 
 class MyRFB(RemoteFrameBuffer):
@@ -211,7 +212,7 @@ def test_widget_traits():
     assert w.max_buffered_frames == 2
     w.max_buffered_frames = 99
     w.max_buffered_frames = 1
-    with raises(Exception):  # TraitError -> min 1
+    with raises(TraitError):  # TraitError -> min 1
         w.max_buffered_frames = 0
 
     assert w.css_width.endswith("px")