Merge branch 'main' into project-formatting

Author: gubaidulinvadim

pyaml/lattice/abstract_impl.py

@@ -173,12 +173,12 @@ class BPMScalarAggregator(ScalarAggregator):
     BPM simulator aggregator
     """
 
-    def __init__(self, ring: at.Lattice):
-        self.__lattice = ring
-        self.__refpts = []
+    def __init__(self, ring:at.Lattice):
+        self.lattice = ring
+        self.refpts = []
 
-    def add_elem(self, elem: at.Element):
-        self.__refpts.append(self.__lattice.index(elem))
+    def add_elem(self,elem:at.Element):
+        self.refpts.append(self.lattice.index(elem))
 
     def set(self, value: NDArray[np.float64]):
         pass
@@ -187,7 +187,7 @@ def set_and_wait(self, value: NDArray[np.float64]):
         pass
 
     def get(self) -> np.array:
-        _, orbit = at.find_orbit(self.__lattice, refpts=self.__refpts)
+        _, orbit = at.find_orbit(self.lattice, refpts=self.refpts)
         return orbit[:, [0, 2]].flatten()
 
     def readback(self) -> np.array:
@@ -200,26 +200,26 @@ def unit(self) -> str:
 # ------------------------------------------------------------------------------
 
 
-class BPMVScalarAggregator(BPMScalarAggregator):
+class BPMHScalarAggregator(BPMScalarAggregator):
     """
     Vertical BPM simulator aggregator
     """
 
     def get(self) -> np.array:
-        _, orbit = at.find_orbit(self.__lattice, refpts=self.__refpts)
+        _, orbit = at.find_orbit(self.lattice, refpts=self.refpts)
         return orbit[:, 0]
 
 
 # ------------------------------------------------------------------------------
 
 
-class BPMHScalarAggregator(BPMScalarAggregator):
+class BPMVScalarAggregator(BPMScalarAggregator):
     """
     Horizontal BPM simulator aggregator
     """
 
     def get(self) -> np.array:
-        _, orbit = at.find_orbit(self.__lattice, refpts=self.__refpts)
+        _, orbit = at.find_orbit(self.lattice, refpts=self.refpts)
         return orbit[:, 2]
 
 

tests/test_arrays.py

@@ -116,13 +116,19 @@ def test_arrays(install_test_package):
 
     # Test BPMs array
 
-    # Using aggragtor
+    # Using aggregator
     pos = sr.design.get_bpms("BPMS").positions.get()
     assert np.abs(pos[0][0] + 7.21154171490481e-05) < 1e-10
     assert np.abs(pos[0][1] - 3.3988843436571406e-05) < 1e-10
     assert np.abs(pos[1][0] - 1.1681211772781844e-04) < 1e-10
     assert np.abs(pos[1][1] - 7.072972488250373e-06) < 1e-10
 
+    # Using aggregator (h and v)
+    pos_h = sr.design.get_bpms("BPMS").h.get()
+    pos_v = sr.design.get_bpms("BPMS").v.get()
+    assert np.all(np.isclose(pos[:, 0], pos_h, rtol=1e-15, atol=1e-15))
+    assert np.all(np.isclose(pos[:, 1], pos_v, rtol=1e-15, atol=1e-15))
+
     # No aggregator
     bpms = []
     for b in sr.design.get_bpms("BPMS"):