diff options
Diffstat (limited to 't/unit/test_simple.py')
| -rw-r--r-- | t/unit/test_simple.py | 5 |
1 files changed, 2 insertions, 3 deletions
diff --git a/t/unit/test_simple.py b/t/unit/test_simple.py index a5cd899a..6a9a9b09 100644 --- a/t/unit/test_simple.py +++ b/t/unit/test_simple.py @@ -91,9 +91,8 @@ class SimpleBase: def test_enter_exit(self): q = self.Queue('test_enter_exit') q.close = Mock() - - assert q.__enter__() is q - q.__exit__() + with q as x: + assert x is q q.close.assert_called_with() def test_qsize(self): |