Improved coverage for lock call context

Author: jtc42

tests/test_core_lock.py

@@ -68,7 +68,6 @@ def test_rlock_block(this_lock):
 
     # Override owner to force acquisition failure
     this_lock._owner = None
-    print(this_lock._owner)
     # Assert not owner
     assert not this_lock._is_owned()
 
@@ -88,22 +87,30 @@ def test_rlock_block(this_lock):
 
     # Release lock, assert no owner
     assert not this_lock._is_owned()
-    
-def test_rlock_acquire_timeout(this_lock):
+
+
+def test_rlock_acquire_timeout_pass(this_lock):
+    # Assert acquisition fails using context manager
+    with this_lock(timeout=0.01) as result:
+        assert result is True
+
+    assert not this_lock.locked()
+
+
+def test_rlock_acquire_timeout_fail(this_lock):
     from labthings.core.exceptions import LockError
 
     # Acquire lock
     assert this_lock.acquire()
 
     # Override owner to force acquisition failure
     this_lock._owner = None
-    print(this_lock._owner)
     # Assert not owner
     assert not this_lock._is_owned()
 
     # Assert acquisition fails using context manager
     with pytest.raises(LockError):
-        with this_lock(timeout=0.01):
+        with this_lock(timeout=0.01) as result:
             pass
 
     # Force ownership