diff options
Diffstat (limited to 't/unit/test_messaging.py')
| -rw-r--r-- | t/unit/test_messaging.py | 7 |
1 files changed, 4 insertions, 3 deletions
diff --git a/t/unit/test_messaging.py b/t/unit/test_messaging.py index f8ed437c..4bd467c2 100644 --- a/t/unit/test_messaging.py +++ b/t/unit/test_messaging.py @@ -1,3 +1,5 @@ +from __future__ import annotations + import pickle import sys from collections import defaultdict @@ -188,9 +190,8 @@ class test_Producer: def test_enter_exit(self): p = self.connection.Producer() p.release = Mock() - - assert p.__enter__() is p - p.__exit__() + with p as x: + assert x is p p.release.assert_called_with() def test_connection_property_handles_AttributeError(self): |