def METHOD_NAME(self):
    ds_name = "test"
    self._master_client.report_dataset_shard_params(
        batch_size=64,
        num_epochs=1,
        dataset_size=10000,
        shuffle=False,
        num_minibatches_per_shard=10,
        dataset_name=ds_name,
        task_type=0,
        storage_type="",
    )
    _, task = self._master_client.get_task(ds_name)
    self._master_client.report_task_result(ds_name, task.task_id, "")
    _, task = self._master_client.get_task(ds_name)
    checkpoint = self._master_client.get_shard_checkpoint(ds_name)
    checkpoint = json.loads(checkpoint)
    self.assertListEqual(checkpoint["doing"][0], [640, 1280])
    self.assertEqual(len(checkpoint["todo"]), 14)
    checkpoint["doing"][0] = [1280, 1600]
    checkpoint = json.dumps(checkpoint)
    self._master_client.report_shard_checkpoint(checkpoint)
    dataset = self._master.task_manager.get_dataset(ds_name)
    task = dataset.todo[0]
    self.assertEqual(task.shard.start, 1280)
    self.assertEqual(task.shard.end, 1600)