a E a&@sXdZddlmZddlmZmZmZmZd ddZee ddd Z e ee d d d Z dS)z(Different utilities for the numpy brains) extract_node) AttributeImportNameNodeNGNcCst|}|j|dS)N)context)rZinfer)srcnoderr >lib/python3.9/site-packages/astroid/brain/brain_numpy_utils.pyinfer_numpy_membersr )r returncCsJ|j}dd||dD}|D]"}d|f|jvs>d|jvr"dSq"dS)a> Returns True if the node is a representation of a numpy module. For example in : import numpy as np x = np.linspace(1, 2) The node is a representation of the numpy module. :param node: node to test :return: True if the node is a representation of the numpy module. cSsg|]}t|tr|qSr ) isinstancer).0xr r r !sz&_is_a_numpy_module..numpy)rNTF)namelookupnames)r Zmodule_nicknameZpotential_import_targettargetr r r _is_a_numpy_modules  r) member_namer r cCsZt|tr.|j|kr.t|jtr.t|jr.dSt|trV|j|krV|jdrVdSdS)z Returns True if the node is a member of numpy whose name is member_name. :param member_name: name of the member :param node: node to test :return: True if the node is a member of numpy TrF) rrattrnameexprrrrroot startswith)rr r r r looks_like_numpy_member-s"  r)N) __doc__Zastroid.builderrZastroid.node_classesrrrrr boolrstrrr r r r  s