Skip to content

Commit abaee68

Browse files
committed
Remove old fft checks
Signed-off-by: Eric Kerfoot <[email protected]>
1 parent 7283385 commit abaee68

File tree

3 files changed

+1
-21
lines changed

3 files changed

+1
-21
lines changed

tests/networks/layers/test_hilbert_transform.py

Lines changed: 0 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -18,8 +18,6 @@
1818
from parameterized import parameterized
1919

2020
from monai.networks.layers import HilbertTransform
21-
from monai.utils import OptionalImportError
22-
from tests.test_utils import SkipIfModule, SkipIfNoModule
2321

2422

2523
def create_expected_numpy_output(input_datum, **kwargs):
@@ -112,7 +110,6 @@ def create_expected_numpy_output(input_datum, **kwargs):
112110
# TESTS CHECKING PADDING, AXIS SELECTION ETC ARE COVERED BY test_detect_envelope.py
113111

114112

115-
@SkipIfNoModule("torch.fft")
116113
class TestHilbertTransformCPU(unittest.TestCase):
117114
@parameterized.expand(TEST_CASES_CPU + TEST_CASES_GPU)
118115
def test_value(self, arguments, image, expected_data, atol):
@@ -121,11 +118,5 @@ def test_value(self, arguments, image, expected_data, atol):
121118
np.testing.assert_allclose(result, expected_data.squeeze(), atol=atol)
122119

123120

124-
@SkipIfModule("torch.fft")
125-
class TestHilbertTransformNoFFTMod(unittest.TestCase):
126-
def test_no_fft_module_error(self):
127-
self.assertRaises(OptionalImportError, HilbertTransform(), torch.randn(1, 1, 10))
128-
129-
130121
if __name__ == "__main__":
131122
unittest.main()

tests/transforms/test_detect_envelope.py

Lines changed: 1 addition & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -18,8 +18,7 @@
1818
from parameterized import parameterized
1919

2020
from monai.transforms import DetectEnvelope
21-
from monai.utils import OptionalImportError
22-
from tests.test_utils import TEST_NDARRAYS, SkipIfModule, SkipIfNoModule, assert_allclose
21+
from tests.test_utils import TEST_NDARRAYS, assert_allclose
2322

2423
n_samples = 500
2524
hann_windowed_sine = np.sin(2 * np.pi * 10 * np.linspace(0, 1, n_samples)) * np.hanning(n_samples)
@@ -114,7 +113,6 @@
114113
TEST_CASE_INVALID_OBJ = [{}, "a string", "__call__"] # method expected to raise exception
115114

116115

117-
@SkipIfNoModule("torch.fft")
118116
class TestDetectEnvelope(unittest.TestCase):
119117
@parameterized.expand(
120118
[
@@ -149,11 +147,5 @@ def test_value_error(self, arguments, image, method):
149147
self.fail("Expected raising method invalid. Should be __init__ or __call__.")
150148

151149

152-
@SkipIfModule("torch.fft")
153-
class TestHilbertTransformNoFFTMod(unittest.TestCase):
154-
def test_no_fft_module_error(self):
155-
self.assertRaises(OptionalImportError, DetectEnvelope(), np.random.rand(1, 10))
156-
157-
158150
if __name__ == "__main__":
159151
unittest.main()

tests/transforms/test_fourier.py

Lines changed: 0 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -20,13 +20,10 @@
2020
from monai.data.synthetic import create_test_image_2d, create_test_image_3d
2121
from monai.transforms import Fourier
2222
from monai.utils.misc import set_determinism
23-
from tests.test_utils import SkipIfNoModule
2423

2524
TEST_CASES = [((128, 64),), ((64, 48, 80),)]
2625

2726

28-
29-
@SkipIfNoModule("torch.fft")
3027
class TestFourier(unittest.TestCase):
3128
def setUp(self):
3229
set_determinism(0)

0 commit comments

Comments
 (0)