We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent b5a4b19 commit cd55b21Copy full SHA for cd55b21
1 file changed
mkl_fft/tests/test_patch.py
@@ -30,14 +30,13 @@
30
31
32
def test_patch():
33
- mkl_fft.restore_numpy_fft()
+ old_module = np.fft.fft.__module__
34
assert not mkl_fft.is_patched()
35
- assert (np.fft.fft.__module__ == "numpy.fft")
36
37
mkl_fft.patch_numpy_fft() # Enable mkl_fft in Numpy
38
assert mkl_fft.is_patched()
39
- assert (np.fft.fft.__module__ == _nfft.fft.__module__)
+ assert np.fft.fft.__module__ == _nfft.fft.__module__
40
41
mkl_fft.restore_numpy_fft() # Disable mkl_fft in Numpy
42
43
+ assert np.fft.fft.__module__ == old_module
0 commit comments