Code used in the AIT of the GRAVITY WFS