desy-ml / cheetah

Fast and differentiable particle accelerator optics simulation for reinforcement learning and optimisation applications.
https://cheetah-accelerator.readthedocs.io
GNU General Public License v3.0
33 stars 13 forks source link

Establish testing test_accelerator.py #39

Closed FelixTheilen closed 1 year ago

FelixTheilen commented 1 year ago

This pull request addresses the issue establish testing #7 .

Every element in the accelerator.py file is tested, by applying this element on a parameter as well as a particle beam and comparing their outputs to the desired outputs. The Elements tested are: -Quadrupole -HorizontalCorrector -VerticalCorrector -Cavity -BPM -Screen -Undulator