def METHOD_NAME(
    parallel: int,
    partitions: Sequence[str],
    current_time: datetime,
    expected: OptimizationSchedule,
) -> None:
    optimize_scheduler = OptimizeScheduler(parallel=parallel)
    with freeze_time(current_time):
        assert optimize_scheduler.get_next_schedule(partitions) == expected