def test_asynchronous_scheduler_prod(self, mock_schedule): scheduler = AsynchronousMachineStepScheduler() scheduler.schedule(1, 'step1') mock_schedule.assert_called_once_with(None, 'machine_task_executor', 1, 'step1')
def test_asynchronous_scheduler(self, mock_schedule): scheduler = AsynchronousMachineStepScheduler() scheduler.schedule(1, 'step1') mock_schedule.assert_called_once_with(1, 'step1')