def METHOD_NAME(self, m_task_execute: MagicMock):
    m_task_execute.return_value = TaskResult(
        id=random.randint(0, 10), status=RunStatus.SUCCESS
    )
    _scheduler = Scheduler(
        project="self",
        version="fdsie8rwe",
        workdir=Path(),
        dataset_uris=["mnist/version/tu788", "mnist/version/tu789"],
        steps=[
            Step(
                name="ppl",
                resources=[{"type": "cpu", "limit": 1, "request": 1}],
                concurrency=1,
                task_num=2,
                needs=[],
            ),
            Step(
                name="cmp",
                resources=[{"type": "cpu", "limit": 1, "request": 1}],
                concurrency=1,
                task_num=1,
                needs=["ppl"],
            ),
        ],
    )
    _results = _scheduler._schedule_all()
    assert m_task_execute.call_count == 3
    self.assertEqual(
        all([_rt.status == RunStatus.SUCCESS for _rt in _results]), True
    )
    m_task_execute.reset_mock()
    _single_result = _scheduler._schedule_one_task("ppl", 0)
    assert m_task_execute.call_count == 1
    self.assertEqual(_single_result.status, RunStatus.SUCCESS)
    m_task_execute.reset_mock()
    _single_result = _scheduler._schedule_one_step("ppl")
    assert m_task_execute.call_count == 2
    self.assertEqual(_single_result.status, RunStatus.SUCCESS)