Added tests for empty array

Author: JeanLucPons

tests/test_arrays.py

@@ -9,7 +9,9 @@
 from pyaml.arrays.element_array import ElementArray
 from pyaml.arrays.magnet_array import MagnetArray
 from pyaml.configuration.factory import Factory
-
+from pyaml.arrays.magnet import Magnet,ConfigModel as MagnetArrayConfigModel
+from pyaml.arrays.bpm import BPM,ConfigModel as BPMArrayConfigModel
+from pyaml.arrays.cfm_magnet import CombinedFunctionMagnet,ConfigModel as CombinedFunctionMagnetConfigModel
 
 @pytest.mark.parametrize(
     "install_test_package",
@@ -220,4 +222,20 @@ def test_arrays(install_test_package):
     magSH1AC = acfm["model_name:SH1A-*"]  # All SH1A magnets (CFMs only)
     assert isinstance(magSH1AC, ElementArray) and len(magSH1AC) == 32
 
+    # Empty arrays
+    emptyMag = Magnet(MagnetArrayConfigModel(name="EmptyMag",elements=[]))
+    emptyMag.fill_array(sr.design) # Attach the array
+    v = sr.design.get_magnets("EmptyMag").strengths.get() # Ensure good attach
+    assert np.shape(v) == (0,)
+
+    emptyBPM = BPM(BPMArrayConfigModel(name="emptyBPM",elements=[]))
+    emptyBPM.fill_array(sr.design) # Attach the array
+    v = sr.design.get_bpms("emptyBPM").positions.get() # Ensure good attach
+    assert np.shape(v) == (0,)
+
+    emptyCFM = CombinedFunctionMagnet(CombinedFunctionMagnetConfigModel(name="emptyCFM",elements=[]))
+    emptyCFM.fill_array(sr.design) # Attach the array
+    v = sr.design.get_cfm_magnets("emptyCFM").strengths.get() # Ensure good attach
+    assert np.shape(v) == (0,)
+
     Factory.clear()