adds use of VisitNotFound

DiamondLightSource · Apr 26, 2023 · 73f31fb · 73f31fb
1 parent 8610920
commit 73f31fb
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 3 deletions.
diff --git a/src/dls_utilpack/visit.py b/src/dls_utilpack/visit.py
@@ -63,6 +63,6 @@ def get_xchem_directory(parent, visit):
     full_path = Path(parent) / subdirectory
 
     if not full_path.is_dir():
-        raise RuntimeError(f"the visit directory {str(full_path)} does not exist")
+        raise VisitNotFound(f"the visit directory {str(full_path)} does not exist")
 
     return str(full_path)
diff --git a/tests/test_visit.py b/tests/test_visit.py
@@ -3,7 +3,11 @@
 
 import pytest
 
-from dls_utilpack.visit import get_xchem_directory, get_xchem_subdirectory
+from dls_utilpack.visit import (
+    VisitNotFound,
+    get_xchem_directory,
+    get_xchem_subdirectory,
+)
 
 # Base class for the tester.
 from tests.base_tester import BaseTester
@@ -59,6 +63,6 @@ async def _main_coroutine(
         assert get_xchem_directory(str(parent), "aa12345-1") == str(full_path)
 
         # Check invalid directory.
-        with pytest.raises(RuntimeError) as excinfo:
+        with pytest.raises(VisitNotFound) as excinfo:
             get_xchem_directory("something", "aa12345-1")
         assert "does not exist" in str(excinfo.value)