U
e5dp @ s d dl mZ d dlmZmZmZ m
Z m Z
d dlmZmZmZmZ d dlmZ d dlmZm Z d dl m Z m!Z" d dl#m#Z$ d dlZ%zd d l&m'Z& W n e(k
r d d l)m'Z& Y nX d
d d d
ddddddddddddddd d d d d d!gZ*d"ed# ed$ Z+d$e Z,ed%Z-d&ed' Z.d(Z/d)e/ Z0d dl1Z1G d*d
d
e1j2Z2G d+d! d!e2Z3d,d- Z4d4d/d0Z5e2 Z6e6j7Z7e6j8Z8e6j9Z9e6j:Z:e6j;Z;e6jZ>e6j?Z?e6j@Z@e6jAZAe6jBZBe6jCZCe6jDZDe6jEZEe6jFZFe6jGZGe6jHZHe6jIZIe6jJZJe6jKZKe6jLZLeMe%d1re%jNe6j7d2 eOd3kre5 dS )5 )warn)logexppieceil)sqrtacoscossin)urandom)SetSequence)
accumulaterepeat)bisectN)sha512Randomseedrandomuniformrandintchoicesample randrangeshuffle
normalvariatelognormvariate expovariatevonmisesvariate gammavariate
triangulargauss betavariate
paretovariateweibullvariategetstatesetstate getrandbitschoices SystemRandom g @ @ ? @5 c s e Zd ZdZd>ddZdd Zd? fdd Z fd
d Z fd d
Zdd Z dd Z
dd Z dde fddZ
dd Zdd Ze de> fdd ZeZd d Zd@d d Zd!d" ZdAddd#d$d%Zd&d' ZdBd*d+Zd,d- Zd.d/ Zd0d1 Zd2d3 Z d4d5 Z d6d7 Z d8d9 Z d:d; Z d}d|jkr qFd|jkr.| j| _ qFd|jkr| j| _ qFqd S )N
_randbelowr( r )__mro____dict___randbelow_with_getrandbitsr: _randbelow_without_getrandbits)clskwargscr7 r7 r8 __init_subclass__g s
zRandom.__init_subclass__r1 c s |dkrt |ttfrt |tr*|dn|}|rBt|d d> nd}tt|D ]}d| |A d@ }qP|t|N }|dkr~dn|}|d krt |tttfrt |tr| }|t |
7 }t |d
}t
| d | _d S ) N zlatin-1r iCB l r1 big)
isinstancestrbytesdecodeordmaplen bytearrayencode_sha512Zdigestint
from_bytessuperr r4 )r5 aversionr6 rA __class__r7 r8 r { s
z Random.seedc s | j t | jfS r3 )VERSIONrT r& r4 r5 rW r7 r8 r& s zRandom.getstatec
s |d }|dkr*|\}}| _ t | nt|dkr|\}}| _ ztdd |D }W n( tk
r| } z
t|W 5 d }~X Y nX t | ntd|| jf d S )Nr r2 r1 c s s | ]}|d V qdS )l Nr7 ).0r6 r7 r7 r8 s z"Random.setstate..z?state with version %s passed to Random.setstate() of version %s)r4 rT r' tuple
ValueError TypeErrorrY )r5 staterV Z
internalstater rW r7 r8 r' s zRandom.setstatec C s | S r3 )r& rZ r7 r7 r8 __getstate__ s zRandom.__getstate__c C s | | d S r3 )r' )r5 r` r7 r7 r8 __setstate__ s zRandom.__setstate__c C s | j d| fS )Nr7 )rX r& rZ r7 r7 r8
__reduce__ s zRandom.__reduce__rC c
C s ||}||krt d|d kr:|dkr2| |S t d||}||krRt d|| }|dkrx|dkrx|| | S |dkrt d|||f ||}||krt d|dkr|| d | } n"|dk r|| d | } nt d| dkrt d||| | S ) Nz!non-integer arg 1 for randrange()r zempty range for randrange()z non-integer stop for randrange()rC z(empty range for randrange() (%d, %d, %d)z non-integer step for randrange()zzero step for randrange())r^ r: )
r5 startstopstep_intZistartZistopwidthZistepnr7 r7 r8 r s4
zRandom.randrangec C s | ||d S NrC )r r5 rU br7 r7 r8 r s zRandom.randintc C s, | j }| }||}||kr(||}q|S r3 )r(
bit_length)r5 ri r( krr7 r7 r8 r= s
z"Random._randbelow_with_getrandbitsc C sn | j }||kr$td || | S |dkr4td|| }|| | }| }||kr^| }qN||| | 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 _warnr^ )r5 ri rR maxsizer Zremlimitro r7 r7 r8 r> s z%Random._randbelow_without_getrandbitsc C s: z| t|}W n tk
r0 tdd Y nX || S )Nz$Cannot choose from an empty sequence)r: rN r^
IndexError)r5 seqir7 r7 r8 r s
z
Random.choicec C s |d krN| j }ttdt|D ]*}||d }|| || ||< ||< q nHt}ttdt|D ]0}|| |d }|| || ||< ||< qdd S rj )r: reversedrangerN rR )r5 r6 r randbelowru jrg r7 r7 r8 r % s zRandom.shufflec C s t |trt|}t |ts$td| j}t|}d| krF|ksPn tdd g| }d}|dkr|dtt |d d 7 }||krt
|}t |D ]0}||| } || ||< ||| d || < qnHt }
|
j
} t |D ]2}||} | |
kr||} q| | || ||< q|S ) Nz>Population must be a sequence or set. For dicts, use list(d).r z,Sample larger than population or is negative r+ r2 rC )rH _Setr] _Sequencer_ r: rN r^ _ceil_loglistrw setadd) r5
populationrn rx ri resultZsetsizeZpoolru ry ZselectedZ selected_addr7 r7 r8 r ; s6 )
z
Random.sample) cum_weightsrn c s | j td krV|d krHt d7 fddtd |D S tt|n|d k rftdtkrztdtd d d fddtd |D S ) N c s g | ]} qS r7 r7 r[ ru )rg ri r r r7 r8
s z"Random.choices..z2Cannot specify both weights and cumulative weightsz3The number of weights does not match the populationrE rC c s$ g | ] } d qS )r r7 r )r r hir r totalr7 r8 r s ) r rN rR _repeatr _accumulater_ r^ _bisect)r5 r Zweightsr rn r7 )rg r r r ri r r r r8 r) s$ zRandom.choicesc C s ||| | S r3 r rk r7 r7 r8 r s zRandom.uniformr r. c C s| | }z |d krdn|| || }W n tk
r@ | Y S X ||krdd| }d| }|| }}||| t|| S )N ?r. )r ZeroDivisionError_sqrt)r5 ZlowZhighmodeurA r7 r7 r8 r! s
zRandom.triangularc C sP | j }| }d| }t|d | }|| d }|t| krqDq||| S )Nr. r r- )r
NV_MAGICCONSTr )r5 musigmar u1u2zZzzr7 r7 r8 r s
zRandom.normalvariatec C s t | ||S r3 )_expr )r5 r r r7 r7 r8 r s zRandom.lognormvariatec C s t d| | S Nr. )r r )r5 Zlambdr7 r7 r8 r s zRandom.expovariatec C s | j }|dkrt| S d| }|td|| }| }tt| }||| }| } | d|| k s| d| t| kr4qq4d| }
|
| d|
| } | } | dkr|t| t }
n|t| t }
|
S )Ngư>r r. )r TWOPIr _cos_pir _acos)r5 r Zkappar sro r r dr qfZu3Zthetar7 r7 r8 r s$
$zRandom.vonmisesvariatec C s~ |dks|dkrt d| j}|dkrtd| d }|t }|| }| }d| k r`dk sdqF qFd| }t|d| | } |t| }
|| | } ||| |
} | t d| dks| t| krF|
| S qFn|dkrtd| | S | }
t| t }||
}|dkr$|d| }
nt|| | }
| }|dkr^||
|d krpqrq|t|
krqrq|
| S d S )Nr z*gammavariate: alpha and beta must be > 0.0r. r, gHz>gP?r/ )r^ r r LOG4r r
SG_MAGICCONST_e)r5 alphabetar ZainvZbbbZcccr r vr6 r ro r rl pr7 r7 r8 r # s@
zRandom.gammavariatec C s` | j }| j}d | _|d krT| t }tdtd| }t|| }t|| | _||| S )Ng r. )r r4 r r r r _sin)r5 r r r r Zx2piZg2radr7 r7 r8 r" h s
z Random.gaussc C s0 | |d}|dkrdS ||| |d S d S )Nr. r r )r )r5 r r yr7 r7 r8 r# s
zRandom.betavariatec C s d| }d|d| S r r )r5 r r r7 r7 r8 r$ s zRandom.paretovariatec C s" d| }|t| d| S r )r r )r5 r r r r7 r7 r8 r% s zRandom.weibullvariate)N)Nr1 )N)N)r r. N)#__name__
__module__ __qualname__rY r9 rB r r& r' ra rb rc rR r r r= BPFr> r: r r r r) r r! r r r r r r" r# r$ r%
__classcell__r7 r7 rW r8 r N s:
,
G
0E5 c @ s4 e Zd Zdd Zdd Zdd Zdd Ze ZZd S )
r* c C s t tddd? t S )NrD rG r2 )rR rS _urandom RECIP_BPFrZ r7 r7 r8 r s zSystemRandom.randomc C s< |dkrt d|d d }tt|d}||d | ? S )Nr z(number of bits must be greater than zerorD rG )r^ rR rS r )r5 rn Znumbytesr6 r7 r7 r8 r( s
zSystemRandom.getrandbitsc O s d S r3 r7 r5 argskwdsr7 r7 r8 r s zSystemRandom.seedc O s t dd S )Nz*System entropy source does not have state.)NotImplementedErrorr r7 r7 r8 _notimplemented s z SystemRandom._notimplementedN) r r r r r( r r r& r' r7 r7 r7 r8 r* s
c C s dd l }t| d|j d}d}d}d}| }t| D ]4} || }
||
7 }||
|
}t|
|}t|
|}q6| } tt| | dddd || } t|| | | }
td
| |
||f d S ) Nr timesr g _Bg _r2 zsec, )endz"avg %g, stddev %g, min %g, max %g
) timeprintr perf_counterrw minmaxroundr )ri funcr r r ZsqsumZsmallestZlargestZt0ru r6 t1ZavgZstddevr7 r7 r8 _test_generator s(
r c C s 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 t | td
d S )Nr7 )r r. )g{Gz?r. )皙?r. )r r, )r r. )g?r. )r. r. )r, r. )g 4@r. )g i@r. ) @r )r r. gUUUUUU?) r r r r r r r" r# r! )Nr7 r7 r8 _test s r fork)Zafter_in_child__main__)r )Pwarningsr rp Zmathr r r r r r r r r r~ r r r r r
r r r osr r _collections_abcr
r| r r} itertoolsr r r r r r _osrQ r ImportErrorZhashlib__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( hasattrregister_at_forkr r7 r7 r7 r8 ( s
{