update ldexp dtypes

Author: Vahid Tavanashad

mkl_umath/tests/test_basic.py

@@ -26,6 +26,7 @@
 import pytest
 import numpy as np
 import mkl_umath._ufuncs as mu
+import mkl_umath._patch as mp
 
 np.random.seed(42)
 
@@ -57,6 +58,7 @@ def get_args(args_str):
     types = mkl_umath.types
     for type_ in types:
         args_str = type_[:type_.find('->')]
+        print(mkl_umath)
         args = get_args(args_str)
         generated_cases[(umath, type_)] = args
 
@@ -82,6 +84,10 @@ def test_umath(case):
 
     assert np.allclose(mkl_res, np_res), f"Results for '{umath}': mkl_res: {mkl_res}, np_res: {np_res}"
 
-def test_cases_count():
-    print("Test cases count:", len(test_cases))
-    assert len(test_cases) > 0, "No test cases found"
+def test_patch():
+    assert not mp.is_patched()
+    mp.use_in_numpy()  # Enable mkl_umath in Numpy
+    assert mp.is_patched()
+
+    mp.restore()  # Disable mkl_umath in Numpy
+    assert not mp.is_patched()