Resuming at 227 for eight, India added 36 runs to the overnight total with Wriddhiman Saha (58 not out) completing his second consecutive half-century of the game.
Tom Latham (31) and Martin Guptill (24) were at the crease for the visitors.
Brief Score:
India: 316 and 263 all out in 76.5 overs (Rohit Sharma 82, Wriddhiman Saha 58 not out Virat Kohli 45; Matt Henry 3/59, Trent Boult 3/38 Mitchell Santner 3/59).