jiuyuan
/
test_ok1
2
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Simplify is_numpy_array as suggested in review

This commit is contained in:
Bruno Oliveira 2018-08-01 07:28:39 -03:00
parent 5003bae0de
commit ad5ddaf55a
1 changed files with 4 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
src/_pytest/python_api.py
View File

@ -531,17 +531,11 @@ def _is_numpy_array(obj):
Return true if the given object is a numpy array. Make a special effort to Return true if the given object is a numpy array. Make a special effort to
avoid importing numpy unless it's really necessary. avoid importing numpy unless it's really necessary.
""" """
import inspect import sys
for cls in inspect.getmro(type(obj)):
if cls.__module__ == "numpy":
try:
import numpy as np
np = sys.modules.get("numpy")
if np is not None:
return isinstance(obj, np.ndarray) return isinstance(obj, np.ndarray)
except ImportError:
pass
return False return False