File tree Expand file tree Collapse file tree 1 file changed +12
-0
lines changed Expand file tree Collapse file tree 1 file changed +12
-0
lines changed Original file line number Diff line number Diff line change 1414# limitations under the License.
1515# ==============================================================================
1616
17+ import inspect
1718import os
1819import pathlib
1920import 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
You can’t perform that action at this time.
0 commit comments