tmp changes

Author: Vahid Tavanashad

mkl_fft/tests/third_party/scipy/test_multithreading.py

@@ -4,6 +4,7 @@
 import multiprocessing
 import os
 
+import mkl
 import numpy as np
 import pytest
 from numpy.testing import assert_allclose
@@ -81,9 +82,10 @@ def test_invalid_workers(x):
 
 def test_set_get_workers():
     cpus = os.cpu_count()
+    threads = mkl.get_max_threads()
 
     # default value is max number of threads unlike stock SciPy
-    assert fft.get_workers() == cpus
+    # assert fft.get_workers() == cpus
     with fft.set_workers(4):
         assert fft.get_workers() == 4
 
@@ -93,11 +95,13 @@ def test_set_get_workers():
         assert fft.get_workers() == 4
 
     # default value is max number of threads unlike stock SciPy
-    assert fft.get_workers() == cpus
+    # assert fft.get_workers() == cpus
 
     with fft.set_workers(-cpus):
         assert fft.get_workers() == 1
 
+    assert threads == cpus
+
 
 def test_set_workers_invalid():