| """Tests for adaptive scheduler.""" | |
| from __future__ import annotations | |
| from dipauglib.schedulers.adaptive import AdaptiveAugmentationScheduler | |
| def test_scheduler_bounds(): | |
| scheduler = AdaptiveAugmentationScheduler() | |
| assert scheduler.d_min <= scheduler.intensity_at(0) <= scheduler.d_max | |
| assert scheduler.d_min <= scheduler.intensity_at(100) <= scheduler.d_max | |
| def test_scheduler_is_monotonic(): | |
| scheduler = AdaptiveAugmentationScheduler() | |
| start = scheduler.intensity_at(0) | |
| middle = scheduler.intensity_at(50) | |
| end = scheduler.intensity_at(100) | |
| assert start < middle < end | |