Qwen-Image-Layered-MRP-MLX/tests/schedulers/test_scheduler_lookup.py
Anthony Wu 6947a23d6c
Scheduler interface implementation. Implement pre-existing linear scheduler only. (#258)
Co-authored-by: Anthony Wu <pls-file-gh-issue@users.noreply.github.com>
Co-authored-by: filipstrand <strand.filip@gmail.com>
2025-10-12 12:20:40 +02:00

18 lines
580 B
Python

import pytest
import mflux.schedulers
def test_scheduler_by_path():
mflux.schedulers.try_import_external_scheduler("mflux.schedulers.linear_scheduler.LinearScheduler")
def test_scheduler_bad_module():
with pytest.raises(mflux.schedulers.SchedulerModuleNotFound):
mflux.schedulers.try_import_external_scheduler("someone.other.project.BarScheduler")
def test_scheduler_bad_classname():
with pytest.raises(mflux.schedulers.SchedulerClassNotFound):
mflux.schedulers.try_import_external_scheduler("mflux.schedulers.linear_scheduler.FooBarScheduler")