Coverage for /var/srv/projects/api.amasfac.comuna18.com/tmp/venv/lib/python3.9/site-packages/numpy/random/_pickle.py: 36%

1from .mtrand import RandomState 

2from ._philox import Philox 

3from ._pcg64 import PCG64, PCG64DXSM 

4from ._sfc64 import SFC64 

5 

6from ._generator import Generator 

7from ._mt19937 import MT19937 

8 

9BitGenerators = {'MT19937': MT19937, 

10 'PCG64': PCG64, 

11 'PCG64DXSM': PCG64DXSM, 

12 'Philox': Philox, 

13 'SFC64': SFC64, 

14 } 

15 

16 

17def __generator_ctor(bit_generator_name='MT19937'): 

18 """ 

19 Pickling helper function that returns a Generator object 

20 

21 Parameters 

22 ---------- 

23 bit_generator_name : str 

24 String containing the core BitGenerator 

25 

26 Returns 

27 ------- 

28 rg : Generator 

29 Generator using the named core BitGenerator 

30 """ 

31 if bit_generator_name in BitGenerators: 

32 bit_generator = BitGenerators[bit_generator_name] 

33 else: 

34 raise ValueError(str(bit_generator_name) + ' is not a known ' 

35 'BitGenerator module.') 

36 

37 return Generator(bit_generator()) 

38 

39 

40def __bit_generator_ctor(bit_generator_name='MT19937'): 

41 """ 

42 Pickling helper function that returns a bit generator object 

43 

44 Parameters 

45 ---------- 

46 bit_generator_name : str 

47 String containing the name of the BitGenerator 

48 

49 Returns 

50 ------- 

51 bit_generator : BitGenerator 

52 BitGenerator instance 

53 """ 

54 if bit_generator_name in BitGenerators: 

55 bit_generator = BitGenerators[bit_generator_name] 

56 else: 

57 raise ValueError(str(bit_generator_name) + ' is not a known ' 

58 'BitGenerator module.') 

59 

60 return bit_generator() 

61 

62 

63def __randomstate_ctor(bit_generator_name='MT19937'): 

64 """ 

65 Pickling helper function that returns a legacy RandomState-like object 

66 

67 Parameters 

68 ---------- 

69 bit_generator_name : str 

70 String containing the core BitGenerator 

71 

72 Returns 

73 ------- 

74 rs : RandomState 

75 Legacy RandomState using the named core BitGenerator 

76 """ 

77 if bit_generator_name in BitGenerators: 

78 bit_generator = BitGenerators[bit_generator_name] 

79 else: 

80 raise ValueError(str(bit_generator_name) + ' is not a known ' 

81 'BitGenerator module.') 

82 

83 return RandomState(bit_generator())