Skip to content

Commit 79b28ad

Browse files
authored
[testing] _onedal_cpu/gpu_supported design test (#1636)
* Update test_patching.py * Update test_patching.py * Update test_patching.py * Update test_patching.py * Update test_patching.py * Update test_patching.py
1 parent 3239e6c commit 79b28ad

File tree

1 file changed

+12
-0
lines changed

1 file changed

+12
-0
lines changed

sklearnex/tests/test_patching.py

Lines changed: 12 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -14,6 +14,7 @@
1414
# limitations under the License.
1515
# ==============================================================================
1616

17+
import inspect
1718
import os
1819
import pathlib
1920
import re
@@ -122,3 +123,14 @@ def test_is_patched_instance(estimator):
122123
unpatched = UNPATCHED_MODELS[estimator]
123124
assert is_patched_instance(patched), f"{patched} is a patched instance"
124125
assert not is_patched_instance(unpatched), f"{unpatched} is an unpatched instance"
126+
127+
128+
@pytest.mark.parametrize("member", ["_onedal_cpu_supported", "_onedal_gpu_supported"])
129+
@pytest.mark.parametrize(
130+
"name",
131+
[i for i in PATCHED_MODELS.keys() if "sklearnex" in PATCHED_MODELS[i].__module__],
132+
)
133+
def test_onedal_supported_member(name, member):
134+
patched = PATCHED_MODELS[name]
135+
sig = str(inspect.signature(getattr(patched, member)))
136+
assert "(self, method_name, *data)" == sig

0 commit comments

Comments
 (0)