Change sys.exit to raise ImportError (#2440)

[etc/vim.git] / tests / util.py

diff --git a/tests/util.py b/tests/util.py
index 1e86a3f3d7873bff36b3dbcea9253f12af63d6b5..e83017f5ad3eba54a98ab78bf560f7e46318dc2d 100644 (file)
--- a/tests/util.py
+++ b/tests/util.py
@@ -1,7 +1,8 @@
 import os
 import unittest
 from pathlib import Path
 import os
 import unittest
 from pathlib import Path

-from typing import List, Tuple, Any
+from typing import Iterator, List, Tuple, Any
+from contextlib import contextmanager

 from functools import partial
 
 import black
 from functools import partial
 
 import black


@@ -72,3 +73,14 @@ def read_data_from_file(file_name: Path) -> Tuple[str, str]:
         # If there's no output marker, treat the entire file as already pre-formatted.
         _output = _input[:]
     return "".join(_input).strip() + "\n", "".join(_output).strip() + "\n"
         # If there's no output marker, treat the entire file as already pre-formatted.
         _output = _input[:]
     return "".join(_input).strip() + "\n", "".join(_output).strip() + "\n"

+
+
+@contextmanager
+def change_directory(path: Path) -> Iterator[None]:
+    """Context manager to temporarily chdir to a different directory."""
+    previous_dir = os.getcwd()
+    try:
+        os.chdir(path)
+        yield
+    finally:
+        os.chdir(previous_dir)