| import pytest |
| from sparsevlm.scheduler import SparsityScheduler, make_scheduler |
|
|
|
|
| def test_bucket_bounds(): |
| s = SparsityScheduler(256, 10, 32) |
| assert s.buckets[0] == 32 |
| assert s.buckets[-1] == 256 |
|
|
|
|
| def test_snap_always_gte(): |
| s = SparsityScheduler(256, 10, 32) |
| for n in range(32, 257, 7): |
| assert s.snap_to_bucket(n) >= n |
|
|
|
|
| def test_snap_in_buckets(): |
| s = SparsityScheduler(256, 10, 32) |
| for n in range(32, 257, 5): |
| assert s.snap_to_bucket(n) in s.buckets |
|
|
|
|
| def test_make_scheduler(): |
| s = make_scheduler(256) |
| assert s.n_vis_max == 256 |
| assert len(s.buckets) > 0 |
|
|
|
|
| def test_bucket_idx(): |
| s = SparsityScheduler(256, 10, 32) |
| for n in [32, 64, 128, 256]: |
| idx = s.get_bucket_idx(n) |
| assert 0 <= idx < len(s.buckets) |
|
|
|
|
| def test_summary(): |
| s = make_scheduler(256) |
| assert isinstance(s.summary(), str) |
| assert len(s.summary()) > 0 |
|
|
