TST: Improve MultiPatcher interface

octue · Jul 27, 2023 · baa6133 · baa6133
1 parent 96b0778
commit baa6133
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/octue/utils/patches.py b/octue/utils/patches.py
@@ -14,12 +14,18 @@ def __enter__(self):
 
         :return list(unittest.mock.MagicMock):
         """
-        return [patch.start() for patch in self.patches]
+        return self.start()
 
     def __exit__(self, *args, **kwargs):
         """Stop the patches.
 
         :return None:
         """
+        self.stop()
+
+    def start(self):
+        return [patch.start() for patch in self.patches]
+
+    def stop(self):
         for patch in self.patches:
             patch.stop()