KL Rahul practicing in tough pitches for the Sri Lanka Test series