def METHOD_NAME(self):
    with contextlib.redirect_stdout(StringIO()):
        trainer, epoch_itr = get_trainer_and_epoch_itr(0, 150, 0, 0)
        trainer.get_train_iterator = MagicMock(return_value=epoch_itr)
        self.patches['os.path.isfile'].return_value = False
        _, epoch_itr = checkpoint_utils.load_checkpoint(self.args_mock, trainer)
        itr = epoch_itr.next_epoch_itr(shuffle=False)
        self.assertEqual(epoch_itr.epoch, 1)
        self.assertEqual(epoch_itr.iterations_in_epoch, 0)
        self.assertEqual(next(itr)['net_input']['src_tokens'][0].item(), 0)