3
\2k @ s d dl mZ d dlmZmZ d dlm Z
m Z m
ZmZmZ d dlmZmZmZmZ d dlm Z d dl m Z m!Z" d dl#m$Z% d dl&Z'd dl(Z)d d
d d d
ddddddddddddddd d d d d gZ*d!e d1 ed# Z+d#e Z,e
d$Z-d%e
d& Z.d'Z/d(e/ Z0d dl1Z1G d)d d e1j2Z2G d*d d e2Z3d+d, Z4d2d.d/Z5e2 Z6e6j7Z7e6j8Z8e6j9Z9e6j:Z:e6j;Z;e6jZ>e6j?Z?e6j@Z@e6jAZAe6jBZBe6jCZCe6jDZDe6jEZEe6jFZFe6jGZGe6jHZHe6jIZIe6jJZJe6jKZKe6jLZLeMd0kre5 dS )3 )warn)
MethodTypeBuiltinMethodType)logexppieceil)sqrtacoscossin)urandom)SetSequence)sha512NRandomseedrandomuniformrandintchoicesample randrangeshuffle
normalvariatelognormvariate expovariatevonmisesvariate gammavariate
triangulargauss betavariate
paretovariateweibullvariategetstatesetstate getrandbitschoices SystemRandom g ?g @g @g ?g @5 c s e Zd ZdZd:ddZd; fdd Z fdd Z fd
d Zd d
Zdd Z dd Z
dde fddZ dd Z
e de> eeefddZdd Zdd&d'Zd(d) Zd*d+ Zd,d- Zd.d/ Z d0d1 Z d2d3 Z d4d5 Z d6d7 Z d8d9 Z! Z"S )?r Nc C s | j | d | _d S )N)r
gauss_next)selfx r1 /usr/lib64/python3.6/random.py__init__W s
zRandom.__init__r, c s |dkrt |ttfrt |tr*|jdn|}|rBt|d d> nd}x"tt|D ]}d| |A d@ }qRW |t|N }|d krd
n|}|dkrt |tttfrt |tr|j }|t |j
7 }t j |d}t
j| d | _d S ) N zlatin-1r iCB l r, big)
isinstancestrbytesdecodeordmaplen bytearrayencode_sha512Zdigestint
from_bytessuperr r. )r/ aversionr0 c) __class__r1 r2 r ` s
z Random.seedc s | j t j | jfS )N)VERSIONrE r% r. )r/ )rI r1 r2 r% s zRandom.getstatec s |d }|dkr*|\}}| _ t j| nt|dkr|\}}| _ ytdd |D }W n( tk
r| } z t|W Y d d }~X nX t j| ntd|| jf d S )Nr r- r, c s s | ]}|d V qdS )r, Nl r1 ).0r0 r1 r1 r2 s z"Random.setstate..z?state with version %s passed to Random.setstate() of version %s)r. rE r& tuple
ValueError TypeErrorrJ )r/ staterG Z
internalstater )rI r1 r2 r& s zRandom.setstatec C s | j S )N)r% )r/ r1 r1 r2 __getstate__ s zRandom.__getstate__c C s | j | d S )N)r& )r/ rQ r1 r1 r2 __setstate__ s zRandom.__setstate__c C s | j f | j fS )N)rI r% )r/ r1 r1 r2
__reduce__ s zRandom.__reduce__r4 c
C s ||}||krt d|d kr:|dkr2| j|S t d||}||krRt d|| }|dkrx|dkrx|| j| S |dkrt d|||f ||}||krt d|dkr|| d | } n"|dk r|| d | } nt d| dkrt d||| j| S ) Nz!non-integer arg 1 for randrange()r zempty range for randrange()z non-integer stop for randrange()r4 z'empty range for randrange() (%d,%d, %d)z non-integer step for randrange()zzero step for randrange())rO
_randbelow)
r/ startstopstep_intZistartZistopwidthZistepnr1 r1 r2 r s4
zRandom.randrangec C s | j ||d S )Nr4 )r )r/ rF br1 r1 r2 r s zRandom.randintc
C s | j }| j}|||ks$|||krN|j } || }
x|
|krH|| }
q6W |
S ||krltd || | S |dkr|td|| } || | } | }
x|
| kr| }
qW ||
| | S )NzUnderlying random() generator does not supply
enough bits to choose from a population range this large.
To remove the range limitation, add a getrandbits() method.r zBoundary cannot be zero)r r'
bit_length_warnrO )
r/ r[ rC maxsizetypeZMethodZ
BuiltinMethodr r' krZremlimitr1 r1 r2 rU s&
zRandom._randbelowc C s: y| j t|}W n tk
r0 tdd Y nX || S )Nz$Cannot choose from an empty sequence)rU r? rO
IndexError)r/ seqir1 r1 r2 r s
z
Random.choicec C s |d krR| j }xttdt|D ]*}||d }|| || ||< ||< q"W nLt}xFttdt|D ]0}|| |d }|| || ||< ||< qjW d S )Nr4 )rU reversedranger? rC )r/ r0 r randbelowrf jrY r1 r1 r2 r s zRandom.shufflec C s( t |trt|}t |ts$td| j}t|}d| koD|kn sRtdd g| }d}|dkr|dtt |d d 7 }||krt
|}xt |D ]0}||| } || ||< ||| d || < qW nRt }
|
j
} xDt |D ]8}||} x| |
kr
||} qW | | || ||< qW |S ) Nz>Population must be a sequence or set. For dicts, use list(d).r z,Sample larger than population or is negative r* r- r4 )r9 _SetrN _SequencerP rU r? rO _ceil_loglistrh setadd) r/
populationra ri r[ resultZsetsizeZpoolrf rj ZselectedZ selected_addr1 r1 r2 r s6
z
Random.sample) cum_weightsra c s | j d krN|d kr>t t fddt|D S ttj|n|d k r^tdttkrvtdt j
d td fddt|D S )Nc s g | ]} qS r1 r1 )rL rf )rY rt r totalr1 r2
c s z"Random.choices..z2Cannot specify both weights and cumulative weightsz3The number of weights does not match the populationr4 c s$ g | ] } d qS )r r1 )rL rf )bisectrv hirt r rw r1 r2 rx l s r7 ) r rC r? rh rq
_itertools
accumulaterP rO _bisectry )r/ rt Zweightsrv ra r1 )rY ry rv rz rt r rw r2 r( W s zRandom.choicesc C s ||| | j S )N)r )r/ rF r\ r1 r1 r2 r s s zRandom.uniform ?c C sx | j }y |d krdn|| || }W n tk
r< |S X ||kr`d| }d| }|| }}||| || d S )Ng ?g ?)r ZeroDivisionError)r/ ZlowZhighmodeurH r1 r1 r2 r y s
zRandom.triangularc C sT | j }x@| }d| }t|d | }|| d }|t| krP qW ||| S )Ng ?g ?g @)r
NV_MAGICCONSTrp )r/ musigmar u1u2zZzzr1 r1 r2 r s
zRandom.normalvariatec C s t | j||S )N)_expr )r/ r r r1 r1 r2 r s zRandom.lognormvariatec C s t d| j | S )Ng ?)rp r )r/ Zlambdr1 r1 r2 r s zRandom.expovariatec C s | j }|dkrt| S d| }|td|| }xN| }tt| }||| }| } | d|| k s~| d| t| kr6P q6W d| }
|
| d|
| } | } | dkr|t| t }
n|t| t }
|
S )Ngư>g ?g ?)r TWOPI_sqrt_cos_pir _acos)r/ r Zkappar srb r r dr qfZu3Zthetar1 r1 r2 r s&
$zRandom.vonmisesvariatec C s |dks|dkrt d| j}|dkrtd| d }|t }|| }x| }d| k o`dk n shqHd| }t|d| | } |t| }
|| | } ||| |
} | t d| dks| t| krH|
| S qHW n|dkr| }
x|
dkr | }
qW t|
| S x| }
t| t }||
}|dkrD|d| }
nt|| | }
| }|dkr|||
|d krP n|t|
krP qW |
| S d S )Ng z*gammavariate: alpha and beta must be > 0.0g ?g @gHz>gP?g @)rO r r LOG4rp r
SG_MAGICCONST_e)r/ alphabetar ZainvZbbbZcccr r vr0 r rb r r\ pr1 r1 r2 r sJ
zRandom.gammavariatec C s` | j }| j}d | _|d krT| t }tdtd| }t|| }t|| | _||| S )Ng @g ?g )r r. r r rp r _sin)r/ r r r r Zx2piZg2radr1 r1 r2 r! ? s
z Random.gaussc C s0 | j |d}|dkrdS ||| j |d S d S )Ng ?r g )r )r/ r r yr1 r1 r2 r" t s
zRandom.betavariatec C s d| j }d|d| S )Ng ?)r )r/ r r r1 r1 r2 r# s zRandom.paretovariatec C s" d| j }|t| d| S )Ng ?)r rp )r/ r r r r1 r1 r2 r$ s zRandom.weibullvariate)N)Nr, )N)N)r~ r N)#__name__
__module__ __qualname__rJ r3 r r% r& rR rS rT rC r r BPFr` _MethodType_BuiltinMethodTyperU r r r r( r r r r r r r r! r" r# r$
__classcell__r1 r1 )rI r2 r G s6
,
:
0H5 c @ s4 e Zd Zdd Zdd Zdd Zdd Ze ZZd S )
r) c C s t jtddd? t S )Nr5 r6 r- )rC rD _urandom RECIP_BPF)r/ r1 r1 r2 r s zSystemRandom.randomc C sP |dkrt d|t|kr$td|d d }tjt|d}||d | ? S )Nr z(number of bits must be greater than zeroz#number of bits should be an integerr5 r6 )rO rC rP rD r )r/ ra Znumbytesr0 r1 r1 r2 r' s zSystemRandom.getrandbitsc O s d S )Nr1 )r/ argskwdsr1 r1 r2 r s zSystemRandom.seedc O s t dd S )Nz*System entropy source does not have state.)NotImplementedError)r/ r r r1 r1 r2 _notimplemented s z SystemRandom._notimplementedN) r r r r r' r r r% r& r1 r1 r1 r2 r) s
c C s dd l }t| d|j d}d}d}d
}|j }x@t| D ]4} || }
||
7 }||
|
}t|
|}t|
|}q8W |j } tt| | dddd || } t|| | | }
td | |
||f d S ) Nr timesg g _Br- zsec, )endz"avg %g, stddev %g, min %g, max %g
g _)timeprintr rh minmaxroundr )r[ funcr r rw ZsqsumZsmallestZlargestZt0rf r0 Zt1ZavgZstddevr1 r1 r2 _test_generator s&
r c C s t | tf t | td t | td t | td
t | td t | td t | td t | td t | td t | td t | td t | td t | td t | td t | td t | td d S )N ?{Gz?皙? @ ?? 4@ i@ @)r r )r r )r r )r r )r r )r r )r r )r r )r r )r r )r r )r r )r r )r r UUUUUU?)r r r ) r r r r r r r! r" r )Nr1 r1 r2 _test s r __main__g )r )Nwarningsr r^ typesr r r r Zmathr rp r r r r r r r ro r
r r r r r r
r osr r _collections_abcr rm r rn Zhashlibr rB itertoolsr{ ry r} __all__r r r r r r Z_randomr r) r r Z_instr r r r r r r r r r( r r r r r r! r" r# r$ r% r& r' r r1 r1 r1 r2 ( sl
Y!