from diffusion import DiffusionConfig


def test_article_schedule_has_ten_rates_and_resolves_the_canvas() -> None:
    config = DiffusionConfig()

    assert config.mask_probabilities == (1.0, 0.9, 0.8, 0.7, 0.6, 0.5, 0.4, 0.3, 0.2, 0.1)
    assert config.masks_after_step(1) == 216
    assert config.masks_after_step(10) == 0