Coverage for /pythoncovmergedfiles/medio/medio/usr/local/lib/python3.11/site-packages/astroid/brain/brain_numpy_ma.py: 80%

1# Licensed under the LGPL: https://www.gnu.org/licenses/old-licenses/lgpl-2.1.en.html 

2# For details: https://github.com/pylint-dev/astroid/blob/main/LICENSE 

3# Copyright (c) https://github.com/pylint-dev/astroid/blob/main/CONTRIBUTORS.txt 

4 

5"""Astroid hooks for numpy ma module.""" 

6 

7from astroid import nodes 

8from astroid.brain.helpers import register_module_extender 

9from astroid.builder import parse 

10from astroid.manager import AstroidManager 

11 

12 

13def numpy_ma_transform() -> nodes.Module: 

14 """ 

15 Infer the call of various numpy.ma functions. 

16 

17 :param node: node to infer 

18 :param context: inference context 

19 """ 

20 return parse(""" 

21 import numpy.ma 

22 def masked_where(condition, a, copy=True): 

23 return numpy.ma.masked_array(a, mask=[]) 

24 

25 def masked_invalid(a, copy=True): 

26 return numpy.ma.masked_array(a, mask=[]) 

27 """) 

28 

29 

30def register(manager: AstroidManager) -> None: 

31 register_module_extender(manager, "numpy.ma", numpy_ma_transform)