More rainy day tests

Author: PokhodenkoSA

dpctl/tests/test_sycl_queue_manager.py

@@ -168,8 +168,10 @@ def test_nested_context_factory_is_empty_list():
 @contextlib.contextmanager
 def _register_nested_context_factory(factory):
     dpctl.nested_context_factories.append(factory)
-    yield
-    dpctl.nested_context_factories.remove(factory)
+    try:
+        yield
+    finally:
+        dpctl.nested_context_factories.remove(factory)
 
 
 def test_register_nested_context_factory_context():
@@ -209,3 +211,19 @@ def factory(_):
             assert in_context
 
         assert not in_context
+
+
+@pytest.mark.parametrize(
+    "factory, exception, match",
+    [
+        (True, TypeError, "object is not callable"),
+        (lambda x: None, AttributeError, "no attribute '__exit__'"),
+    ],
+)
+def test_nested_context_factory_exception_if_wrong_factory(
+    factory, exception, match
+):
+    with pytest.raises(exception, match=match):
+        with _register_nested_context_factory(factory):
+            with dpctl.device_context("opencl:gpu:0"):
+                pass