Skip to content

Commit ac39115

Browse files
committed
Refine test for PAN
1 parent 40d97a3 commit ac39115

File tree

1 file changed

+9
-0
lines changed

1 file changed

+9
-0
lines changed

tests/models/test_pan.py

Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,4 @@
1+
import pytest
12
import segmentation_models_pytorch as smp
23

34
from tests.models import base
@@ -37,3 +38,11 @@ def test_interpolation(self):
3738
assert model_2.decoder.gau2.align_corners is None
3839
assert model_2.decoder.gau3.interpolation_mode == "bicubic"
3940
assert model_2.decoder.gau3.align_corners is None
41+
42+
with pytest.warns(DeprecationWarning):
43+
smp.create_model(
44+
self.test_model_type,
45+
self.test_encoder_name,
46+
upscale_mode="bicubic",
47+
)
48+
assert model_2.decoder.gau1.interpolation_mode == "bicubic"

0 commit comments

Comments
 (0)