Fix formating

Author: JeanLucPons

tests/test_arrays.py

@@ -4,14 +4,18 @@
 import pytest
 
 from pyaml.accelerator import Accelerator
+from pyaml.arrays.bpm import BPM
+from pyaml.arrays.bpm import ConfigModel as BPMArrayConfigModel
 from pyaml.arrays.bpm_array import BPMArray
+from pyaml.arrays.cfm_magnet import CombinedFunctionMagnet
+from pyaml.arrays.cfm_magnet import ConfigModel as CombinedFunctionMagnetConfigModel
 from pyaml.arrays.cfm_magnet_array import CombinedFunctionMagnetArray
 from pyaml.arrays.element_array import ElementArray
+from pyaml.arrays.magnet import ConfigModel as MagnetArrayConfigModel
+from pyaml.arrays.magnet import Magnet
 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",
@@ -223,19 +227,21 @@ def test_arrays(install_test_package):
     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
+    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
+    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
+    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()