mirror of
https://github.com/open-mmlab/mmsegmentation.git
synced 2025-06-03 22:03:48 +08:00
14 lines
376 B
Python
14 lines
376 B
Python
|
from mmseg.models.utils import make_divisible
|
||
|
|
|
||
|
|
|
||
|
|
def test_make_divisible():
|
||
|
|
# test with min_value = None
|
||
|
|
assert make_divisible(10, 4) == 12
|
||
|
|
assert make_divisible(9, 4) == 12
|
||
|
|
assert make_divisible(1, 4) == 4
|
||
|
|
|
||
|
|
# test with min_value = 8
|
||
|
|
assert make_divisible(10, 4, 8) == 12
|
||
|
|
assert make_divisible(9, 4, 8) == 12
|
||
|
|
assert make_divisible(1, 4, 8) == 8
|