Magic Mock Python Test Train