fixes skip test cases if QUICKTEST
@@ -16,6 +16,7 @@ from parameterized import parameterized
from monai.networks.blocks import FCN, MCFCN
from monai.networks.nets import AHNet
+from tests.utils import skip_if_quick
TEST_CASE_FCN_1 = [
{"out_channels": 3, "upsample_mode": "transpose"},
@@ -124,6 +125,7 @@ class TestAHNET(unittest.TestCase):
TEST_CASE_AHNET_3D_3,
]
)
+ @skip_if_quick
def test_ahnet_shape(self, input_param, input_data, expected_shape):
net = AHNet(**input_param)
net.eval()
@@ -134,6 +136,7 @@ class TestAHNET(unittest.TestCase):
class TestAHNETWithPretrain(unittest.TestCase):
@parameterized.expand([TEST_CASE_AHNET_3D_WITH_PRETRAIN_1, TEST_CASE_AHNET_3D_WITH_PRETRAIN_2])
+ @skip_if_quick
def test_ahnet_shape(self, input_param, input_data, expected_shape, fcn_input_param):
net = AHNet(**input_param)
net2d = FCN(**fcn_input_param)
