Rabada puts himself through the paces as South Africa weigh Test return