Merge pull request #170 from yannachen/load_robot

Using config.get() to get robot

src/haven/instrument/robot.py

@@ -106,9 +106,11 @@ def load_robot_coros(config=None):
     # Load PV's from config
     if config is None:
         config = load_config()
-    robots = config["robot"]
-    for name, cfg in robots.items():
-        yield make_device(Robot, name=name, labels={"robots"}, prefix=cfg["prefix"])
+    robots = config.get("robot")
+
+    if robots is not None:
+        for name, cfg in robots.items():
+            yield make_device(Robot, name=name, labels={"robots"}, prefix=cfg["prefix"])
 
 
 def load_robot(config=None):

src/haven/tests/test_robot.py

@@ -1,3 +1,6 @@
+import pytest
+from ophydregistry.exceptions import ComponentNotFound
+
 from haven.instrument import Robot, load_robot
 
 
@@ -17,6 +20,18 @@ def test_load_robot(sim_registry):
     assert rbt.prefix == "255idAustin"
 
 
+def test_load_no_robot(sim_registry):
+    load_robot(config={})
+
+    # Test the robot is not in config
+    result = pytest.raises(ComponentNotFound, sim_registry.findall, label="robots")
+
+    # Assert that the expected exception is raised
+    assert 'Could not find components matching: label="robots", name="None"' in str(
+        result.value
+    )
+
+
 # -----------------------------------------------------------------------------
 # :author:    Yanna Chen
 # :email:     [email protected]